improve prompt and interface

app/interface.py

@@ -47,14 +47,14 @@ def create_interface():
     with gr.Blocks() as interface:
         gr.Markdown("# Flashcard Studio")
         gr.Markdown(
-            "Make flashcards from uploaded files or directly input text. Choose your preferred output format."
+            "Make flashcards from uploaded files or directly input text. Choose CSV or JSON output format."
         )
         with gr.Tab("Upload File"):
             with gr.Row():
                 with gr.Column():
                     file_input = gr.File(
                         label="Upload a File",
-                        file_types=['.pdf', '.txt', '.md']
+                        file_types=['.txt', '.md']
                     )
                     format_selector = gr.Radio(
                         choices=["CSV", "JSON"],
@@ -106,8 +106,7 @@ def create_interface():
             """
             ---
             **Notes:**
-            - Supported file types: `.pdf`, `.txt`, `.md`.
-            - Ensure that the input text is clear and well-structured for optimal flashcard extraction.
+            - Supported file types: `.txt`, `.md`.
             """
         )
 

app/pipeline.py

@@ -10,7 +10,7 @@ class Pipeline:
     def __init__(self, model_name: str = "Qwen/Qwen2.5-7B-Instruct"):
         self.torch_pipe = pipeline(
             "text-generation",
-            "Qwen/Qwen2.5-7B-Instruct",
+            model_name,
             torch_dtype="auto",
             device_map="auto"
         )
@@ -20,6 +20,8 @@ class Pipeline:
             {"role": "system", "content": """You are an expert flashcard creator.
             - You ALWAYS include a single knowledge item per flashcard.
             - You ALWAYS respond in valid JSON format.
+            - You ALWAYS make flashcards accurate and comprehensivce.
+            - If the text includes code snippets, you consider snippets a knowledge item testing the user's understanding of how to write the code and what it does.
 
             Format responses like the example below.
 
@@ -51,4 +53,3 @@ class Pipeline:
             return torch.device("mps")
         else:
             return torch.device("cpu")
-