Update app.py

app.py

@@ -1,10 +1,41 @@
 import gradio as gr
 from groq import Groq
 import os
+import matplotlib.pyplot as plt
+import json
+from PIL import Image
+import requests
+from io import BytesIO
 
 # Initialize Groq client
 client = Groq(api_key=os.environ["GROQ_API_KEY"])
 
+def generate_visual_response(subject, topic):
+    """
+    Generate a visual response based on the subject and topic.
+    """
+    if subject == "Math":
+        # Example: Generate a simple plot for Math
+        plt.figure()
+        plt.plot([0, 1, 2, 3, 4], [0, 1, 4, 9, 16], label="y = x^2")
+        plt.title(f"Math Graph: {topic}")
+        plt.xlabel("x")
+        plt.ylabel("y")
+        plt.legend()
+        plt.savefig("math_plot.png")
+        return "math_plot.png"
+    
+    elif subject == "Science":
+        # Example: Fetch an image from Unsplash for Science
+        response = requests.get(f"https://source.unsplash.com/400x300/?{topic}")
+        img = Image.open(BytesIO(response.content))
+        img.save("science_image.png")
+        return "science_image.png"
+    
+    # Add more subjects as needed
+    else:
+        return None
+
 def generate_tutor_output(subject, difficulty, student_input):
     prompt = f"""
     You are an expert tutor in {subject} at the {difficulty} level. 
@@ -22,7 +53,7 @@ def generate_tutor_output(subject, difficulty, student_input):
         messages=[
             {
                 "role": "system",
-                "content": "You are the world's best AI tutor, renowned for your ability to explain complex concepts in an engaging, clear, and memorable way and giving math examples. Your expertise in {subject} is unparalleled, and you're adept at tailoring your teaching to {difficulty} level students. Your goal is to not just impart knowledge, but to inspire a love for learning and critical thinking.",
+                "content": f"You are the world's best AI tutor, renowned for your ability to explain complex concepts in an engaging, clear, and memorable way and giving math examples. Your expertise in {subject} is unparalleled, and you're adept at tailoring your teaching to {difficulty} level students. Your goal is to not just impart knowledge, but to inspire a love for learning and critical thinking.",
             },
             {
                 "role": "user",
@@ -33,7 +64,10 @@ def generate_tutor_output(subject, difficulty, student_input):
         max_tokens=1000,
     )
 
-    return completion.choices[0].message.content
+    output = completion.choices[0].message.content
+    visual_response = generate_visual_response(subject, student_input)
+    
+    return output, visual_response
 
 with gr.Blocks() as demo:
     gr.Markdown("# 🎓 Your AI Tutor by Farhan")
@@ -61,6 +95,7 @@ with gr.Blocks() as demo:
             lesson_output = gr.Markdown(label="Lesson")
             question_output = gr.Markdown(label="Comprehension Question")
             feedback_output = gr.Markdown(label="Feedback")
+            visual_output = gr.Image(label="Visual Explanation")
     
     gr.Markdown("""
     ### How to Use
@@ -72,17 +107,17 @@ with gr.Blocks() as demo:
     6. Feel free to ask follow-up questions or explore new topics!
     """)
     
-    def process_output(output):
+    def process_output(output, visual_response):
         try:
-            parsed = eval(output)
-            return parsed["lesson"], parsed["question"], parsed["feedback"]
+            parsed = json.loads(output)
+            return parsed["lesson"], parsed["question"], parsed["feedback"], visual_response
         except:
-            return "Error parsing output", "No question available", "No feedback available"
+            return "Error parsing output", "No question available", "No feedback available", None
     
     submit_button.click(
-        fn=lambda s, d, i: process_output(generate_tutor_output(s, d, i)),
+        fn=lambda s, d, i: process_output(*generate_tutor_output(s, d, i)),
         inputs=[subject, difficulty, student_input],
-        outputs=[lesson_output, question_output, feedback_output]
+        outputs=[lesson_output, question_output, feedback_output, visual_output]
     )
 
 if __name__ == "__main__":