Update app.py

app.py

@@ -18,15 +18,20 @@ import os
 import json
 import logging
 from typing import Dict, Any, Tuple
+from dotenv import load_dotenv
+import markdown2
+
+# Load environment variables
+load_dotenv()
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # Initialize Groq client
 try:
-    client = Groq(api_key=os.environ["GROQ_API_KEY"])
-except KeyError:
-    logging.error("GROQ_API_KEY not found in environment variables")
+    client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+except Exception as e:
+    logging.error(f"Error initializing Groq client: {str(e)}")
     raise EnvironmentError("Please set the GROQ_API_KEY environment variable")
 
 def transcribe_audio(audio):
@@ -52,56 +57,62 @@ def transcribe_audio(audio):
         return f"Error in transcription: {str(e)}"
 
 def generate_tutor_output(subject: str, difficulty: str, student_input: str) -> Dict[str, str]:
-    # Define enhanced math topics
-    math_topics = {
-        "quadratic equation": "solving quadratic equations, including methods like factoring, using the quadratic formula, and completing the square",
-        "pythagorean theorem": "the Pythagorean theorem, its proof, and applications in geometry and trigonometry",
-        "calculus": "fundamental concepts of calculus, including limits, derivatives, and integrals",
-        "linear algebra": "basics of linear algebra, including vectors, matrices, and linear transformations",
-        "statistics": "key concepts in statistics, such as probability distributions, hypothesis testing, and regression analysis"
+    # Define enhanced topics for all subjects
+    topics = {
+        "math": {
+            "quadratic equation": "solving quadratic equations, including methods like factoring, using the quadratic formula, and completing the square",
+            "pythagorean theorem": "the Pythagorean theorem, its proof, and applications in geometry and trigonometry",
+            "calculus": "fundamental concepts of calculus, including limits, derivatives, and integrals",
+            "linear algebra": "basics of linear algebra, including vectors, matrices, and linear transformations",
+            "statistics": "key concepts in statistics, such as probability distributions, hypothesis testing, and regression analysis"
+        },
+        "science": {
+            "photosynthesis": "the process of photosynthesis in plants, including light-dependent and light-independent reactions",
+            "newton's laws": "Newton's laws of motion and their applications in classical mechanics",
+            "periodic table": "the structure and organization of the periodic table of elements",
+            "dna replication": "the process of DNA replication and its importance in cell division",
+            "climate change": "the causes and effects of climate change, including global warming and its impact on ecosystems"
+        },
+        # Add more subjects and topics as needed
     }
     
-    # Check if it's a math topic and create an enhanced prompt
-    is_math_topic = subject.lower() == "math"
+    # Check if it's a specific topic and create an enhanced prompt
     enhanced_topic = None
-    for topic, description in math_topics.items():
-        if topic.lower() in student_input.lower():
-            enhanced_topic = description
+    for subject_topics in topics.values():
+        for topic, description in subject_topics.items():
+            if topic.lower() in student_input.lower():
+                enhanced_topic = description
+                break
+        if enhanced_topic:
             break
 
-    # Create the prompt based on whether it's a math topic or not
-    if is_math_topic and enhanced_topic:
+    # Create the prompt based on whether it's a specific topic or not
+    if enhanced_topic:
         prompt = f"""
-        You are an expert math tutor specializing in {enhanced_topic} at the {difficulty} level. The student has asked about: "{student_input}"
+        You are an expert tutor specializing in {enhanced_topic} at the {difficulty} level. The student has asked about: "{student_input}"
         Please provide your response in the following format:
-
         1. Lesson: A comprehensive lesson based on national educational standards on the topic (3-4 paragraphs), including its historical context.
-
         2. Example: A detailed step-by-step example problem with a full solution. Structure it as follows:
            - Start with "Example Problem:" followed by the question.
            - Then, list the steps to solve the problem, each on a new line starting with "Step 1:", "Step 2:", etc.
            - Finally, provide the answer starting with "Answer:".
-
         3. Real-World Application: A challenging real-world application of this concept, demonstrating its practical use.
-
-        Format your entire response as a JSON object with keys: "lesson", "example", "real_world_problem".
+        4. Quiz: Create a short quiz (3 multiple-choice questions) to test the student's understanding.
+        Format your entire response as a JSON object with keys: "lesson", "example", "real_world_problem", "quiz".
         Ensure that the content for each key is a single string, with line breaks where appropriate.
         """
     else:
         prompt = f"""
         You are an expert tutor in {subject} at the {difficulty} level. The student has provided the following input: "{student_input}"
         Please generate your response in the following format:
-
         1. Lesson: Provide a descriptive, long, and engaging lesson on the topic (2-3 paragraphs).
-
         2. Example: Present an example problem related to the topic. Structure it as follows:
            - Start with "Example Problem:" followed by the question.
            - Then, list the steps to solve the problem, each on a new line starting with "Step 1:", "Step 2:", etc.
            - Finally, provide the answer starting with "Answer:".
-
         3. Real-World Application: Describe a real-world problem that can be solved using the concepts from the lesson.
-
-        Format your entire response as a JSON object with keys: "lesson", "example", "real_world_problem".
+        4. Quiz: Create a short quiz (3 multiple-choice questions) to test the student's understanding.
+        Format your entire response as a JSON object with keys: "lesson", "example", "real_world_problem", "quiz".
         Ensure that the content for each key is a single string, with line breaks where appropriate.
         """
 
@@ -118,7 +129,7 @@ def generate_tutor_output(subject: str, difficulty: str, student_input: str) ->
                 }
             ],
             model="llama3-groq-70b-8192-tool-use-preview",
-            max_tokens=1000,
+            max_tokens=2000,
         )
         # Parse the JSON string into a dictionary before returning
         return json.loads(completion.choices[0].message.content)
@@ -126,15 +137,16 @@ def generate_tutor_output(subject: str, difficulty: str, student_input: str) ->
         logging.error(f"Error generating tutor output: {str(e)}")
         return {"error": f"Failed to generate tutor output: {str(e)}"}
 
-def process_output(output: Dict[str, Any]) -> Tuple[str, str, str]:
+def process_output(output: Dict[str, Any]) -> Tuple[str, str, str, str]:
     try:
-        lesson = str(output.get("lesson", "No lesson available"))
-        example = str(output.get("example", "No example available"))
-        real_world = str(output.get("real_world_problem", "No real-world problem available"))
-        return lesson, example, real_world
+        lesson = markdown2.markdown(str(output.get("lesson", "No lesson available")))
+        example = markdown2.markdown(str(output.get("example", "No example available")))
+        real_world = markdown2.markdown(str(output.get("real_world_problem", "No real-world problem available")))
+        quiz = markdown2.markdown(str(output.get("quiz", "No quiz available")))
+        return lesson, example, real_world, quiz
     except Exception as e:
         logging.error(f"Error processing output: {str(e)}")
-        return str(e), "", ""
+        return str(e), "", "", ""
 
 def create_interface() -> gr.Blocks:
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -164,16 +176,17 @@ def create_interface() -> gr.Blocks:
 
             with gr.Column(scale=3):
                 transcription_output = gr.Textbox(label="Transcribed Audio (if provided)")
-                lesson_output = gr.Markdown(label="Lesson")
-                example_output = gr.Markdown(label="Example")
-                real_world_output = gr.Markdown(label="Real-World Application")
+                lesson_output = gr.HTML(label="Lesson")
+                example_output = gr.HTML(label="Example")
+                real_world_output = gr.HTML(label="Real-World Application")
+                quiz_output = gr.HTML(label="Quiz")
 
         gr.Markdown("""
         ### How to Use
         1. Select a subject from the dropdown.
         2. Choose your difficulty level.
         3. Enter the topic or question you'd like to explore, or use the microphone to speak your question.
-        4. Click 'Teach Me' to receive a personalized lesson, example, and real-world application.
+        4. Click 'Teach Me' to receive a personalized lesson, example, real-world application, and quiz.
         5. Review the AI-generated content to enhance your learning.
         6. Use the 'Clear' button to reset all fields and start a new query.
         7. Feel free to ask follow-up questions or explore new topics!
@@ -201,25 +214,25 @@ def create_interface() -> gr.Blocks:
                 logging.info(f"Processing input: subject={subject}, difficulty={difficulty}, student_input={student_input}")
                 
                 tutor_output = generate_tutor_output(subject, difficulty, student_input)
-                lesson, example, real_world = process_output(tutor_output)
-                return transcribed_text, lesson, example, real_world
+                lesson, example, real_world, quiz = process_output(tutor_output)
+                return transcribed_text, lesson, example, real_world, quiz
             except Exception as e:
                 logging.error(f"Error in process_input: {str(e)}")
-                return str(e), "Error generating lesson", "Error generating example", "Error generating real-world problem"
+                return str(e), "Error generating lesson", "Error generating example", "Error generating real-world problem", "Error generating quiz"
 
         def clear_outputs():
-            return [""] * 4  # Clear all four output fields
+            return [""] * 5  # Clear all five output fields
 
         submit_button.click(
             fn=process_input,
             inputs=[subject, difficulty, student_input, audio_input],
-            outputs=[transcription_output, lesson_output, example_output, real_world_output]
+            outputs=[transcription_output, lesson_output, example_output, real_world_output, quiz_output]
         )
 
         clear_button.click(
             fn=clear_outputs,
             inputs=[],
-            outputs=[transcription_output, lesson_output, example_output, real_world_output]
+            outputs=[transcription_output, lesson_output, example_output, real_world_output, quiz_output]
         )
 
     return demo