Update app.py

app.py

@@ -5,6 +5,7 @@ import json
 
 # Initialize Groq client
 client = Groq(api_key=os.environ["GROQ_API_KEY"])
+print("API Key:", os.environ.get("GROQ_API_KEY"))  # Debug print
 
 def generate_tutor_output(subject, grade, student_input, model):
     prompt = f"""
@@ -19,22 +20,31 @@ def generate_tutor_output(subject, grade, student_input, model):
     Format your response as a JSON object with keys: "lesson", "question", "options", "correct_answer", "feedback"
     """
 
-    completion = client.chat.completions.create(
-        messages=[
-            {
-                "role": "system",
-                "content": f"You are a fun, creative AI tutor for {grade} graders, expert in {subject}. You explain concepts in a simple, exciting way with relatable examples (like math problems for their age). Your goal is to spark curiosity and help students practice what they learn!",
-            },
-            {
-                "role": "user",
-                "content": prompt,
-            }
-        ],
-        model=model,
-        max_tokens=1200,
-    )
-
-    return completion.choices[0].message.content
+    try:
+        completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "system",
+                    "content": f"You are a fun, creative AI tutor for {grade} graders, expert in {subject}. You explain concepts in a simple, exciting way with relatable examples (like math problems for their age). Your goal is to spark curiosity and help students practice what they learn!",
+                },
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            model=model,
+            max_tokens=1200,
+        )
+        return completion.choices[0].message.content
+    except Exception as e:
+        print(f"Groq API Error: {str(e)}")  # Debug print
+        return json.dumps({
+            "lesson": f"Error: Could not generate lesson. API error: {str(e)}",
+            "question": "No question available",
+            "options": [],
+            "correct_answer": "",
+            "feedback": "No feedback available due to API error"
+        })
 
 def check_answer(selected_answer, correct_answer):
     if selected_answer == correct_answer:
@@ -61,7 +71,7 @@ with gr.Blocks(title="Learn & Practice 🚀") as demo:
             model_select = gr.Dropdown(
                 [
                     "mixtral-8x7b-32768",
-                    "qwen-2.5-coder-32b",
+                    "llama3-8b-8192",  # Fallback model
                     "qwen-2.5-32b"
                 ],
                 label="AI Tutor Model",
@@ -95,6 +105,7 @@ with gr.Blocks(title="Learn & Practice 🚀") as demo:
     """)
 
     def process_output(output):
+        print(f"Raw API Output: {output}")  # Debug print
         try:
             parsed = json.loads(output)
             options = [f"{k}. {v}" for k, v in zip(["a", "b", "c", "d"], parsed["options"])]
@@ -105,9 +116,10 @@ with gr.Blocks(title="Learn & Practice 🚀") as demo:
                 parsed["correct_answer"],
                 parsed["feedback"]
             )
-        except:
+        except Exception as e:
+            print(f"JSON Parsing Error: {str(e)}")  # Debug print
             return (
-                "Error generating lesson",
+                f"Error parsing response: {str(e)}",
                 "No question available",
                 [],
                 "",
@@ -115,6 +127,7 @@ with gr.Blocks(title="Learn & Practice 🚀") as demo:
             )
 
     def update_interface(subject, grade, student_input, model):
+        print(f"Selected Model: {model}")  # Debug print
         output = generate_tutor_output(subject, grade, student_input, model)
         lesson, question, options, correct_answer, feedback = process_output(output)
         return (