Update app.py

app.py

@@ -429,6 +429,8 @@ def extract_and_fix_json(text, subject="General"):
         dict: Parsed JSON or generic plan
     """
     if not text or not isinstance(text, str):
+        if st.session_state.show_debug:
+            st.warning("⚠️ No text to parse in extract_and_fix_json")
         return None
     
     # Store raw response for debug
@@ -443,31 +445,50 @@ def extract_and_fix_json(text, subject="General"):
     text = re.sub(r'```javascript\s*', '', text, flags=re.IGNORECASE)
     text = re.sub(r'```python\s*', '', text, flags=re.IGNORECASE)
     
+    # Show cleaned text in debug
+    if st.session_state.show_debug:
+        with st.expander("🧹 Cleaned Text (first 500 chars)", expanded=False):
+            st.code(text[:500] + "..." if len(text) > 500 else text)
+    
     # Find JSON boundaries
     start_idx = text.find('{')
     end_idx = text.rfind('}')
     
     if start_idx == -1 or end_idx == -1 or end_idx <= start_idx:
+        if st.session_state.show_debug:
+            st.error(f"❌ No JSON found. Start: {start_idx}, End: {end_idx}")
         return None
     
     json_str = text[start_idx:end_idx+1]
     
-    # DEBUG: Show what we're trying to parse
+    # Show JSON string in debug
     if st.session_state.show_debug:
         with st.expander("🔍 JSON String Being Parsed", expanded=False):
-            st.code(json_str[:2000] + "..." if len(json_str) > 2000 else json_str)
+            st.code(json_str[:1000] + "..." if len(json_str) > 1000 else json_str)
     
     try:
         # First try: Direct parse
-        return json.loads(json_str)
+        parsed_json = json.loads(json_str)
+        if st.session_state.show_debug:
+            st.success("✅ JSON parsed successfully on first try")
+        return parsed_json
     except json.JSONDecodeError as e:
+        if st.session_state.show_debug:
+            st.warning(f"⚠️ First parse failed: {str(e)}")
+            st.info("Trying to fix JSON...")
+        
         # Second try: Apply fixes
         json_str = fix_json_string(json_str)
         try:
-            return json.loads(json_str)
-        except json.JSONDecodeError:
+            parsed_json = json.loads(json_str)
+            if st.session_state.show_debug:
+                st.success("✅ JSON parsed successfully after fixes")
+            return parsed_json
+        except json.JSONDecodeError as e2:
+            if st.session_state.show_debug:
+                st.error(f"❌ Second parse failed: {str(e2)}")
             # Third try: Create generic plan
-            return create_generic_plan(subject, 60)
+            return None
 
 def fix_json_string(json_str):
     """
@@ -643,7 +664,10 @@ def generate_study_plan(api_key, **kwargs):
     """
     Generate study plan using Gemini
     """
-    genai.configure(api_key=api_key)
+    if not api_key:
+        st.error("❌ No API key provided")
+        return create_generic_plan(kwargs.get('subject', 'General'), 
+                                  kwargs.get('days_available', 60))
     
     # Extract parameters
     subject = kwargs.get('subject', 'General Learning')
@@ -690,11 +714,23 @@ Return valid JSON with this exact structure:
   "success_metrics": ["Metric 1", "Metric 2"]
 }}
 
-IMPORTANT: Make sure topics_allocation values are NUMBERS (not strings like "10 hours").
-Include ALL {weeks} weeks in the weekly_schedule array."""
+IMPORTANT: 
+1. Make sure topics_allocation values are NUMBERS (not strings like "10 hours")
+2. Include ALL {weeks} weeks in the weekly_schedule array
+3. Return ONLY JSON, no additional text
+4. Make tasks specific and actionable for {subject}"""
 
     try:
+        # Configure API
+        genai.configure(api_key=api_key)
         model = genai.GenerativeModel('gemini-2.5-flash')
+        
+        # Show prompt in debug mode
+        if st.session_state.show_debug:
+            with st.expander("📝 Prompt Sent to AI", expanded=False):
+                st.code(prompt[:1000] + "..." if len(prompt) > 1000 else prompt)
+        
+        # Generate response
         response = model.generate_content(
             prompt,
             generation_config=genai.GenerationConfig(
@@ -702,15 +738,34 @@ Include ALL {weeks} weeks in the weekly_schedule array."""
                 temperature=0.7
             )
         )
+        
         raw_text = response.text
         st.session_state.raw_response = raw_text
         
+        # Show raw response in debug mode
+        if st.session_state.show_debug:
+            with st.expander("🔍 Raw AI Response", expanded=False):
+                st.text_area("AI Response", raw_text, height=200, key="ai_response_raw")
+        
+        # Extract JSON
         plan = extract_and_fix_json(raw_text, subject)
         
         if plan:
+            # Show success in debug mode
+            if st.session_state.show_debug:
+                st.success("✅ JSON successfully extracted")
+                with st.expander("📋 Extracted Plan Preview", expanded=False):
+                    st.json({
+                        "subject": plan.get('subject'),
+                        "total_weeks": plan.get('total_weeks'),
+                        "weekly_schedule_length": len(plan.get('weekly_schedule', [])),
+                        "first_week_focus": plan.get('weekly_schedule', [{}])[0].get('focus', 'N/A') if plan.get('weekly_schedule') else 'N/A'
+                    })
+            
             # Post-process the plan
             plan['generated_at'] = datetime.now().isoformat()
             plan['total_days'] = days_available
+            plan['user_level'] = current_level
             
             # Initialize week tracking
             for week in plan.get('weekly_schedule', []):
@@ -728,9 +783,16 @@ Include ALL {weeks} weeks in the weekly_schedule array."""
             
             return plan
         else:
+            st.warning("⚠️ Could not extract valid JSON from AI response")
+            if st.session_state.show_debug:
+                st.info("Falling back to generic plan")
             return create_generic_plan(subject, days_available)
     
     except Exception as e:
+        st.error(f"❌ AI Generation Error: {str(e)}")
+        if st.session_state.show_debug:
+            with st.expander("🔍 Error Details", expanded=False):
+                st.exception(e)
         return create_generic_plan(subject, days_available)
 
 def generate_weekly_test(api_key, week_number, weekly_tasks, subject):
@@ -1124,16 +1186,46 @@ tab1, tab2, tab3, tab4, tab5, tab6 = st.tabs([
     "📤 Export"
 ])
 
+
 # ============================================
-# TAB 1: DEFINE GOAL - SIMPLIFIED
+# TAB 1: DEFINE GOAL
 # ============================================
 
 with tab1:
     st.markdown('<h2 class="sub-header">Define Your Learning Goal</h2>', unsafe_allow_html=True)
     
+    # Debug Mode Toggle
+    col_debug1, col_debug2 = st.columns([1, 3])
+    with col_debug1:
+        st.session_state.show_debug = st.checkbox("🔍 Debug Mode", value=False)
+    
+    # API Status
+    with col_debug2:
+        if not st.session_state.api_key:
+            st.error("⚠️ **API Key Missing:** Set GEMINI_API_KEY in Hugging Face Secrets")
+        else:
+            st.success("✅ **API Key Loaded**")
+    
+    # Test API Button
+    if st.session_state.api_key and st.session_state.show_debug:
+        if st.button("Test API Connection", type="secondary", key="test_api"):
+            try:
+                genai.configure(api_key=st.session_state.api_key)
+                model = genai.GenerativeModel('gemini-2.5-flash')
+                test_response = model.generate_content("Say 'API is working'", 
+                    generation_config=genai.GenerationConfig(max_output_tokens=10))
+                if test_response.text:
+                    st.success(f"✅ **API Connected:** {test_response.text}")
+                else:
+                    st.error("❌ No response from API")
+            except Exception as e:
+                st.error(f"❌ **API Error:** {str(e)}")
+    
+    st.markdown("---")
+    
     # Option 1: Upload Existing Plan
     st.info("📤 **Option 1: Upload Existing Plan**")
-    uploaded_file = st.file_uploader("Upload saved study plan (JSON)", type=['json'])
+    uploaded_file = st.file_uploader("Upload saved study plan (JSON)", type=['json'], key="upload_plan")
     
     if uploaded_file is not None and not st.session_state.plan_loaded:
         try:
@@ -1142,8 +1234,8 @@ with tab1:
             st.session_state.subject = plan['subject']
             st.session_state.plan_loaded = True
             st.success(f"✅ Plan loaded: {plan['subject']}")
-        except:
-            st.error("❌ Invalid plan file")
+        except Exception as e:
+            st.error(f"❌ Invalid plan file: {str(e)}")
     
     st.markdown("---")
     
@@ -1156,7 +1248,8 @@ with tab1:
         subject = st.text_input(
             "What do you want to learn?",
             placeholder="e.g., Data Science, Python, Web Development...",
-            help="Enter the subject or topic"
+            help="Enter the subject or topic",
+            key="subject_input"
         )
         
         col1a, col2a = st.columns(2)
@@ -1165,7 +1258,8 @@ with tab1:
                 "Hours per day:",
                 min_value=1,
                 max_value=8,
-                value=2
+                value=2,
+                key="hours_slider"
             )
         
         with col2a:
@@ -1173,30 +1267,35 @@ with tab1:
                 "Days available:",
                 min_value=7,
                 max_value=365,
-                value=60
+                value=60,
+                key="days_slider"
             )
         
         study_days = st.multiselect(
             "Study days:",
             ["Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"],
-            default=["Mon", "Tue", "Wed", "Thu", "Fri"]
+            default=["Mon", "Tue", "Wed", "Thu", "Fri"],
+            key="study_days_select"
         )
         
     with col2:
         current_level = st.selectbox(
             "Your level:",
             ["Beginner", "Intermediate", "Advanced"],
-            index=0
+            index=0,
+            key="level_select"
         )
     
+    st.markdown("---")
+    
     # Generate Plan Button
-    if st.button("🚀 Generate AI Study Plan", type="primary", use_container_width=True):
+    if st.button("🚀 Generate AI Study Plan", type="primary", use_container_width=True, key="generate_plan_btn"):
         if not st.session_state.api_key:
-            st.error("⚠️ API Key Required")
+            st.error("⚠️ **API Key Required:** Please set GEMINI_API_KEY in Hugging Face Secrets.")
         elif not subject:
-            st.error("⚠️ Please enter a subject")
+            st.error("⚠️ Please enter a subject/topic")
         elif not study_days:
-            st.error("⚠️ Please select study days")
+            st.error("⚠️ Please select at least one study day")
         else:
             with st.spinner("🤖 AI is creating your personalized study plan..."):
                 try:
@@ -1216,34 +1315,67 @@ with tab1:
                         'study_days': study_days
                     }
                     
+                    # Show plan data in debug mode
+                    if st.session_state.show_debug:
+                        with st.expander("📋 Plan Data Sent to AI", expanded=True):
+                            st.json(plan_data)
+                    
                     # Generate plan
                     plan = generate_study_plan(st.session_state.api_key, **plan_data)
                     
                     if plan:
                         st.session_state.study_plan = plan
                         st.session_state.subject = subject
+                        st.session_state.current_week = 1
+                        st.session_state.weekly_streak = 0
+                        st.session_state.longest_weekly_streak = 0
                         st.session_state.plan_loaded = True
                         
-                        # Show debug info
-                        if st.session_state.show_debug:
-                            with st.expander("🔍 Raw AI Response", expanded=True):
-                                st.text_area("Full Response", st.session_state.raw_response, height=300)
-                        
                         st.success("✅ Study plan generated successfully!")
                         st.balloons()
                         
                         # Show plan summary
                         st.info(f"📅 **Schedule:** {len(study_days)} days/week ({', '.join(study_days)}) × {hours_per_day} hours/day")
-                        st.info(f"📊 **Duration:** {plan.get('total_weeks', 0)} weeks")
+                        st.info(f"📊 **Duration:** {plan.get('total_weeks', 0)} weeks ({days_available} days)")
                         
                         # Show learning quality
-                        st.info(f"🎯 **Learning Quality:** {learning_quality['quality_score']:.0f}/100")
+                        if learning_quality['is_optimal']:
+                            st.success(f"🎯 **Learning Quality:** {learning_quality['quality_score']:.0f}/100 (Excellent)")
+                        else:
+                            st.warning(f"📊 **Learning Quality:** {learning_quality['quality_score']:.0f}/100 (Needs improvement)")
+                        
+                        st.caption(f"Coverage: {learning_quality['coverage_percentage']:.0f}% | Total hours: {learning_quality['total_hours']}h")
                         
                     else:
-                        st.error("❌ Failed to generate plan")
+                        st.error("❌ Failed to generate plan. Check debug mode for details.")
                         
                 except Exception as e:
                     st.error(f"❌ Error: {str(e)}")
+                    if st.session_state.show_debug:
+                        with st.expander("🔍 Error Details", expanded=True):
+                            st.exception(e)
+    
+    # Debug Info Section
+    if st.session_state.show_debug and st.session_state.raw_response:
+        st.markdown("---")
+        st.subheader("🔍 Debug Information")
+        
+        col_debug_a, col_debug_b = st.columns(2)
+        
+        with col_debug_a:
+            st.metric("Raw Response Length", f"{len(st.session_state.raw_response)} chars")
+        
+        with col_debug_b:
+            if st.session_state.study_plan:
+                plan_type = "AI Generated" if st.session_state.study_plan.get('generated_at') else "Generic"
+                st.metric("Plan Type", plan_type)
+        
+        with st.expander("📋 Raw AI Response", expanded=False):
+            st.text_area("Full Response", st.session_state.raw_response, height=300, key="raw_response_display")
+        
+        if st.session_state.study_plan:
+            with st.expander("📊 Parsed Plan Structure", expanded=False):
+                st.json(st.session_state.study_plan)
 
 # ============================================
 # TAB 2: STUDY PLAN