Update app.py

app.py

@@ -1,7 +1,5 @@
-# app.py
-
 import gradio as gr
-from utils import generate_tutor_output
+from utils import generate_tutor_output, generate_visual_answer
 
 with gr.Blocks() as demo:
     gr.Markdown("# 🎓 Your AI Tutor by Farhan")
@@ -47,18 +45,9 @@ with gr.Blocks() as demo:
     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)
-            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)),
+        fn=lambda s, d, i, m: (generate_tutor_output(s, d, i, m), generate_visual_answer(i)),
         inputs=[subject, difficulty, student_input, model],
         outputs=[lesson_output, question_output, feedback_output, image_output]
     )