Update app.py

app.py

@@ -1,8 +1,11 @@
 import gradio as gr
+from pathlib import Path
 from sentence_transformers import CrossEncoder
 import numpy as np
 from time import perf_counter
-from backend.semantic_search import table, retriever
+from pydantic import BaseModel, Field
+from phi.agent import Agent
+from phi.model.groq import Groq
 import os
 import logging
 
@@ -10,19 +13,59 @@ import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# API Key setup (optional, depending on retriever needs)
+# API Key setup
 api_key = os.getenv("GROQ_API_KEY")
 if not api_key:
-    gr.Warning("GROQ_API_KEY not found. Set it in 'Repository secrets' if required.")
+    gr.Warning("GROQ_API_KEY not found. Set it in 'Repository secrets'.")
     logger.error("GROQ_API_KEY not found.")
 else:
     os.environ["GROQ_API_KEY"] = api_key
 
+# Pydantic Model for Quiz Structure
+class QuizItem(BaseModel):
+    question: str = Field(..., description="The quiz question")
+    choices: list[str] = Field(..., description="List of 4 multiple-choice options")
+    correct_answer: str = Field(..., description="The correct choice (e.g., 'C1')")
+
+class QuizOutput(BaseModel):
+    items: list[QuizItem] = Field(..., description="List of 10 quiz items")
+
+# Initialize Agents
+groq_agent = Agent(model=Groq(model="llama3-70b-8192", api_key=api_key), markdown=True)
+
+quiz_generator = Agent(
+    name="Quiz Generator",
+    role="Generates structured quiz questions and answers",
+    instructions=[
+        "Create 10 questions with 4 choices each based on the provided topic and documents.",
+        "Use the specified difficulty level (easy, average, hard) to adjust question complexity.",
+        "Ensure questions are derived only from the provided documents.",
+        "Return the output in a structured format using the QuizOutput Pydantic model.",
+        "Each question should have a unique correct answer from the choices (labeled C1, C2, C3, C4)."
+    ],
+    model=Groq(id="llama3-70b-8192", api_key=api_key),
+    response_model=QuizOutput,
+    markdown=True
+)
+
 VECTOR_COLUMN_NAME = "vector"
 TEXT_COLUMN_NAME = "text"
+proj_dir = Path.cwd()
+
+# Calling functions from backend (assuming they exist)
+from backend.semantic_search import table, retriever
 
-def retrieve_documents(topic):
-    gr.Warning('Retrieving documents may take a moment. Please wait.', duration=30)
+def generate_quiz_data(question_difficulty, topic, documents_str):
+    prompt = f"""Generate a quiz with {question_difficulty} difficulty on topic '{topic}' using only the following documents:\n{documents_str}"""
+    try:
+        response = quiz_generator.run(prompt)
+        return response.content
+    except Exception as e:
+        logger.error(f"Failed to generate quiz: {e}")
+        return None
+
+def retrieve_and_generate_quiz(question_difficulty, topic):
+    gr.Warning('Generating quiz may take 1-2 minutes. Please wait.', duration=60)
     top_k_rank = 10
     documents = []
 
@@ -37,21 +80,34 @@ def retrieve_documents(topic):
     sim_scores_argsort = list(reversed(np.argsort(cross_scores)))
     documents = [documents[idx] for idx in sim_scores_argsort[:top_k_rank]]
 
-    elapsed_time = perf_counter() - document_start
-    return f"Retrieved {len(documents)} documents in {elapsed_time:.2f} seconds:\n\n" + "\n".join(documents)
+    documents_str = '\n'.join(documents)
+    quiz_data = generate_quiz_data(question_difficulty, topic, documents_str)
+    if not quiz_data or not quiz_data.items:
+        return "Error: Failed to generate quiz."
+
+    # Display only the questions as plain text
+    questions = [f"{i}. {item.question}" for i, item in enumerate(quiz_data.items[:10], 1)]
+    return "\n".join(questions)
 
-with gr.Blocks(title="Document Retrieval Test") as APP:
+with gr.Blocks(title="Quiz Generator Test") as QUIZBOT:
     with gr.Row():
-        gr.Markdown("# Document Retrieval Test")
+        gr.Markdown("# Quiz Generator Test")
+
+    topic = gr.Textbox(label="Enter Topic", placeholder="Write any topic from 9th Science CBSE")
+    difficulty_radio = gr.Radio(["easy", "average", "hard"], label="How difficult should the quiz be?")
     
-    topic = gr.Textbox(label="Enter Topic", placeholder="Enter a topic to search")
-    output = gr.Textbox(label="Retrieved Documents", interactive=False)
+    generate_quiz_btn = gr.Button("Generate Quiz")
+    quiz_output = gr.Textbox(label="Quiz Questions", interactive=False)
 
-    submit_btn = gr.Button("Retrieve Documents")
-    submit_btn.click(fn=retrieve_documents, inputs=[topic], outputs=[output])
+    # Register the click event without @ decorator
+    generate_quiz_btn.click(
+        fn=retrieve_and_generate_quiz,
+        inputs=[difficulty_radio, topic],
+        outputs=[quiz_output]
+    )
 
 if __name__ == "__main__":
-    APP.queue().launch(server_name="0.0.0.0", server_port=7860)
+    QUIZBOT.queue().launch(server_name="0.0.0.0", server_port=7860)
 # import gradio as gr
 # from pathlib import Path
 # from tempfile import NamedTemporaryFile