Update app.py

app.py

@@ -1,35 +1,31 @@
+# app.py
+
 import gradio as gr
-from utils import generate_tutor_output  # Import the function from utils.py
+from utils import generate_tutor_output
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🎓 MistralMind: A precision AI for personalized learning and development")
-    
+    gr.Markdown("# 🎓 Your AI Tutor by Farhan")
+
     with gr.Row():
         with gr.Column(scale=2):
-            # Subject Dropdown
             subject = gr.Dropdown(
                 ["Math", "Science", "History", "Literature", "Code", "AI"], 
                 label="Subject", 
                 info="Choose the subject of your lesson"
             )
-            
-            # Difficulty Level Radio
             difficulty = gr.Radio(
                 ["Beginner", "Intermediate", "Advanced"], 
                 label="Difficulty Level", 
                 info="Select your proficiency level"
             )
-            
-            # Student Input Textbox
             student_input = gr.Textbox(
                 placeholder="Type your query here...", 
                 label="Your Input", 
                 info="Enter the topic you want to learn"
             )
 
-            # Model Dropdown - Updated to match models in utils.py
             model = gr.Dropdown(
-                ["OpenAI", "Meta-Llama-3.1-8B-Instruct", "Meta-Llama-3.3-70B", "Mixtral 8x7B"], 
+                ["OpenAI", "LLAMA3 70B", "Mixtral 8x7B"], 
                 label="LLM", 
                 info="Choose the language model"
             )
@@ -40,28 +36,31 @@ with gr.Blocks() as demo:
             lesson_output = gr.Markdown(label="Lesson")
             question_output = gr.Markdown(label="Comprehension Question")
             feedback_output = gr.Markdown(label="Feedback")
+            image_output = gr.Image(label="Visual Answer", elem_id="image-output")
     
     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.
-    4. Click 'Generate Lesson' to receive a personalized lesson, question, and feedback.
+    4. Click 'Generate Lesson' to receive a personalized lesson, question, feedback, and a visual answer.
     5. Review the AI-generated content to enhance your learning.
     6. Feel free to ask follow-up questions or explore new topics!
     """)
-    
+
+    # Function to process and extract the JSON response
     def process_output(output):
         try:
-            parsed = eval(output)  # Parse the output as JSON
+            parsed = eval(output)
             return parsed["lesson"], parsed["question"], parsed["feedback"]
         except:
             return "Error parsing output", "No question available", "No feedback available"
-    
+
+    # Define the interaction between user inputs and generated outputs
     submit_button.click(
         fn=lambda s, d, i, m: process_output(generate_tutor_output(s, d, i, m)),
         inputs=[subject, difficulty, student_input, model],
-        outputs=[lesson_output, question_output, feedback_output]
+        outputs=[lesson_output, question_output, feedback_output, image_output]
     )
 
 if __name__ == "__main__":