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")
-