Update app.py

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
 import google.generativeai as genai
 
-# Simple test function to show FULL raw AI responses
+# Test function to get COMPLETE AI responses
 def test_ai_responses():
-    st.title("🤖 AI Response Debugger - RAW OUTPUT ONLY")
+    st.title("🤖 AI Response Debugger - COMPLETE OUTPUT")
     
     # Get API Key
     api_key = st.text_input("Enter Gemini API Key:", type="password")
@@ -16,125 +16,110 @@ def test_ai_responses():
         with tab1:
             st.subheader("Test Study Plan Generation")
             
-            subject = st.text_input("Subject:", "Data Science", key="plan_subject")
-            days_available = st.slider("Days Available:", 7, 365, 60, key="plan_days")
-            hours_per_day = st.slider("Hours per Day:", 1, 8, 2, key="plan_hours")
+            subject = st.text_input("Subject:", "Machine Learning", key="plan_subject")
             
             if st.button("Generate Test Study Plan", key="plan_btn"):
-                with st.spinner("Generating..."):
-                    weeks = max(1, days_available // 7)
-                    
-                    prompt = f"""Create a detailed {weeks}-week study plan for: {subject}
-
-Study schedule: {hours_per_day} hours/day, 5 days/week (Monday to Friday)
-
-IMPORTANT: Return JSON ONLY with this exact structure:
-
-{{
-  "subject": "{subject}",
-  "total_weeks": {weeks},
-  "daily_hours": {hours_per_day},
+                with st.spinner("Generating COMPLETE plan (this may take a moment)..."):
+                    # SIMPLER prompt - let AI structure it naturally
+                    prompt = """Create a complete 4-week study plan for Machine Learning.
+Return the plan as valid JSON with this structure:
+{
+  "subject": "Machine Learning",
+  "total_weeks": 4,
+  "daily_hours": 2,
   "study_days": ["Mon", "Tue", "Wed", "Thu", "Fri"],
-  "topics_allocation": {{"Topic1": 10, "Topic2": 20, "Topic3": 30}},
+  "topics_allocation": {"Python Basics": 10, "Statistics": 15, "ML Algorithms": 25, "Projects": 20},
   "weekly_schedule": [
-    {{
+    {
       "week": 1,
-      "focus": "Week focus title",
+      "focus": "Week 1 focus",
       "objectives": ["Objective 1", "Objective 2"],
       "daily_tasks": [
-        ["Task 1 for Monday", "Task 2 for Monday"],
-        ["Task 1 for Tuesday", "Task 2 for Tuesday"],
-        ["Task 1 for Wednesday", "Task 2 for Wednesday"],
-        ["Task 1 for Thursday", "Task 2 for Thursday"],
-        ["Task 1 for Friday", "Task 2 for Friday"]
+        ["Task 1 Monday", "Task 2 Monday"],
+        ["Task 1 Tuesday", "Task 2 Tuesday"],
+        ["Task 1 Wednesday", "Task 2 Wednesday"],
+        ["Task 1 Thursday", "Task 2 Thursday"],
+        ["Task 1 Friday", "Task 2 Friday"]
       ],
-      "day_focus": ["Monday: Introduction", "Tuesday: Practice", "Wednesday: Deep Dive", "Thursday: Review", "Friday: Project"],
-      "week_summary": "Brief summary",
+      "day_focus": ["Monday: Intro", "Tuesday: Practice", "Wednesday: Deep Dive", "Thursday: Review", "Friday: Project"],
+      "week_summary": "Week 1 summary",
       "resources": ["Resource 1", "Resource 2"],
-      "milestone": "Weekly milestone"
-    }}
+      "milestone": "Complete Week 1"
+    }
   ],
   "study_tips": ["Tip 1", "Tip 2"],
   "success_metrics": ["Metric 1", "Metric 2"]
-}}"""
+}
+
+Make sure to include ALL 4 weeks in the weekly_schedule array."""
 
                     try:
                         model = genai.GenerativeModel('gemini-2.5-flash')
                         response = model.generate_content(
                             prompt,
                             generation_config=genai.GenerationConfig(
-                                max_output_tokens=2000,
-                                temperature=0.7
-                            )
+                max_output_tokens=4000,  # INCREASED for complete response
+                temperature=0.7
+            )
                         )
                         
                         raw_text = response.text
-                        st.subheader("FULL RAW AI RESPONSE:")
-                        st.text_area("COPY EVERYTHING FROM HERE:", raw_text, height=600)
+                        st.subheader("COMPLETE RAW AI RESPONSE:")
                         
-                        # Show length info
-                        st.info(f"Response length: {len(raw_text)} characters")
+                        # Show in expander for better viewing
+                        with st.expander("📋 Click to view FULL response", expanded=True):
+                            st.text_area("COPY EVERYTHING BELOW:", raw_text, height=800, key="plan_raw")
                         
-                        # Just show the raw text - NO PARSING ATTEMPT
-                        st.markdown("---")
-                        st.subheader("📋 What to do:")
-                        st.write("1. Copy ALL the text from the box above")
-                        st.write("2. Paste it in the chat for me to analyze")
-                        st.write("3. I will create the correct JSON extraction function")
+                        st.info(f"✅ Response length: {len(raw_text)} characters")
                         
+                        if len(raw_text) < 500:
+                            st.warning("⚠️ Response seems too short! The AI might have stopped early.")
+                            
                     except Exception as e:
                         st.error(f"Error: {str(e)}")
         
         with tab2:
             st.subheader("Test Question Generation")
             
-            week_num = st.number_input("Week Number:", 1, 12, 1, key="test_week")
-            test_subject = st.text_input("Test Subject:", "Data Science", key="test_subject")
-            
             if st.button("Generate Test Questions", key="test_btn"):
                 with st.spinner("Generating test questions..."):
-                    prompt = f"""Create 5 multiple choice questions for Week {week_num} of {test_subject}.
-
-Return JSON:
-{{
+                    prompt = """Create 3 multiple choice questions for Machine Learning.
+Return valid JSON with this structure:
+{
   "questions": [
-    {{
-      "question": "Clear, complete question?",
-      "options": {{
-        "A": "Complete option A text with full explanation",
-        "B": "Complete option B text with full explanation",
-        "C": "Complete option C text with full explanation",
-        "D": "Complete option D text with full explanation"
-      }},
+    {
+      "question": "What is supervised learning?",
+      "options": {
+        "A": "Learning with labeled data",
+        "B": "Learning without labeled data",
+        "C": "Reinforcement learning",
+        "D": "Deep learning"
+      },
       "correct_answer": "A",
-      "explanation": "Brief explanation of correct answer"
-    }}
+      "explanation": "Supervised learning uses labeled data to train models."
+    }
   ]
-}}"""
+}
+
+Include exactly 3 questions."""
 
                     try:
                         model = genai.GenerativeModel('gemini-2.5-flash')
                         response = model.generate_content(
                             prompt,
                             generation_config=genai.GenerationConfig(
-                                max_output_tokens=1200,
-                                temperature=0.7
-                            )
+                max_output_tokens=2000,
+                temperature=0.7
+            )
                         )
                         
                         raw_text = response.text
-                        st.subheader("FULL RAW AI RESPONSE:")
-                        st.text_area("COPY EVERYTHING FROM HERE:", raw_text, height=600)
+                        st.subheader("COMPLETE RAW AI RESPONSE:")
                         
-                        # Show length info
-                        st.info(f"Response length: {len(raw_text)} characters")
+                        with st.expander("📋 Click to view FULL response", expanded=True):
+                            st.text_area("COPY EVERYTHING BELOW:", raw_text, height=600, key="test_raw")
                         
-                        # Just show the raw text - NO PARSING ATTEMPT
-                        st.markdown("---")
-                        st.subheader("📋 What to do:")
-                        st.write("1. Copy ALL the text from the box above")
-                        st.write("2. Paste it in the chat for me to analyze")
-                        st.write("3. I will create the correct JSON extraction function")
+                        st.info(f"✅ Response length: {len(raw_text)} characters")
                         
                     except Exception as e:
                         st.error(f"Error: {str(e)}")