Update app.py

app.py

@@ -24,8 +24,9 @@ class EducationalAgent:
                     },
                     {"role": "user", "content": prompt},
                 ],
-                model="mixtral-8x7b-32768",
-                max_tokens=1000,
+                model="llama3-groq-8b-8192-tool-use-preview",
+                max_tokens=4000,
+                temperature=0.7,
             )
             return completion.choices[0].message.content
         except Exception as e:
@@ -56,10 +57,15 @@ def process_query(query, subject, grade_level):
         return "Invalid subject selected.", "", ""
     
     # Parse the response
+    lesson, question, feedback = "No lesson available", "No question available", "No feedback available"
     parts = response.split("\n\n")
-    lesson = parts[0].replace("Lesson: ", "") if len(parts) > 0 else "No lesson available"
-    question = parts[1].replace("Question: ", "") if len(parts) > 1 else "No question available"
-    feedback = parts[2].replace("Feedback: ", "") if len(parts) > 2 else "No feedback available"
+    for part in parts:
+        if part.startswith("Lesson:"):
+            lesson = part.replace("Lesson:", "").strip()
+        elif part.startswith("Question:"):
+            question = part.replace("Question:", "").strip()
+        elif part.startswith("Feedback:"):
+            feedback = part.replace("Feedback:", "").strip()
     
     return lesson, question, feedback