Update app.py

app.py

@@ -1,11 +1,9 @@
 import streamlit as st
-import json
 import google.generativeai as genai
-import re
 
-# Test function to capture raw AI responses
+# Simple test function to show FULL raw AI responses
 def test_ai_responses():
-    st.title("🤖 AI Response Debugger")
+    st.title("🤖 AI Response Debugger - RAW OUTPUT ONLY")
     
     # Get API Key
     api_key = st.text_input("Enter Gemini API Key:", type="password")
@@ -18,22 +16,17 @@ def test_ai_responses():
         with tab1:
             st.subheader("Test Study Plan Generation")
             
-            subject = st.text_input("Subject:", "Data Science")
-            goal_type = st.selectbox("Goal Type:", ["Skill/Topic Completion", "Certification Preparation"])
-            days_available = st.slider("Days Available:", 7, 365, 60)
-            hours_per_day = st.slider("Hours per Day:", 1, 8, 2)
-            study_days = st.multiselect("Study Days:", ["Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"], 
-                                       default=["Mon", "Tue", "Wed", "Thu", "Fri"])
+            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")
             
-            if st.button("Generate Test Study Plan"):
+            if st.button("Generate Test Study Plan", key="plan_btn"):
                 with st.spinner("Generating..."):
                     weeks = max(1, days_available // 7)
-                    study_days_count = len(study_days)
                     
                     prompt = f"""Create a detailed {weeks}-week study plan for: {subject}
-Goal Type: {goal_type}
-Level: Beginner
-Study schedule: {hours_per_day} hours/day, {study_days_count} days/week ({', '.join(study_days)})
+
+Study schedule: {hours_per_day} hours/day, 5 days/week (Monday to Friday)
 
 IMPORTANT: Return JSON ONLY with this exact structure:
 
@@ -41,24 +34,24 @@ IMPORTANT: Return JSON ONLY with this exact structure:
   "subject": "{subject}",
   "total_weeks": {weeks},
   "daily_hours": {hours_per_day},
-  "study_days": {json.dumps(study_days)},
-  "topics_allocation": {{"Topic1": X, "Topic2": X, "Topic3": X}},
+  "study_days": ["Mon", "Tue", "Wed", "Thu", "Fri"],
+  "topics_allocation": {{"Topic1": 10, "Topic2": 20, "Topic3": 30}},
   "weekly_schedule": [
     {{
       "week": 1,
       "focus": "Week focus title",
       "objectives": ["Objective 1", "Objective 2"],
       "daily_tasks": [
-        ["Task 1 for Monday", "Task 2 for Monday", "Task 3 for Monday"],
+        ["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 3 for Wednesday"],
-        ["Task 1 for Thursday"],
+        ["Task 1 for Wednesday", "Task 2 for Wednesday"],
+        ["Task 1 for Thursday", "Task 2 for Thursday"],
         ["Task 1 for Friday", "Task 2 for Friday"]
       ],
       "day_focus": ["Monday: Introduction", "Tuesday: Practice", "Wednesday: Deep Dive", "Thursday: Review", "Friday: Project"],
-      "week_summary": "Brief summary of what will be learned this week",
+      "week_summary": "Brief summary",
       "resources": ["Resource 1", "Resource 2"],
-      "milestone": "Weekly milestone achievement"
+      "milestone": "Weekly milestone"
     }}
   ],
   "study_tips": ["Tip 1", "Tip 2"],
@@ -76,30 +69,29 @@ IMPORTANT: Return JSON ONLY with this exact structure:
                         )
                         
                         raw_text = response.text
-                        st.subheader("Raw AI Response:")
-                        st.text_area("Copy this for debugging:", raw_text, height=400)
+                        st.subheader("FULL RAW AI RESPONSE:")
+                        st.text_area("COPY EVERYTHING FROM HERE:", raw_text, height=600)
+                        
+                        # Show length info
+                        st.info(f"Response length: {len(raw_text)} characters")
+                        
+                        # 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")
                         
-                        # Try to extract JSON
-                        st.subheader("Extracted JSON Attempt:")
-                        extracted_json = extract_and_fix_json_test(raw_text, subject)
-                        if extracted_json:
-                            st.success("✅ JSON successfully extracted!")
-                            st.json(extracted_json)
-                        else:
-                            st.error("❌ Failed to extract JSON")
-                            st.text("Cleaned text for JSON extraction:")
-                            st.text(clean_json_text(raw_text))
-                            
                     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)
-            test_subject = st.text_input("Test Subject:", "Data Science")
+            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"):
+            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}.
 
@@ -131,146 +123,21 @@ Return JSON:
                         )
                         
                         raw_text = response.text
-                        st.subheader("Raw AI Response:")
-                        st.text_area("Copy this for debugging:", raw_text, height=400)
+                        st.subheader("FULL RAW AI RESPONSE:")
+                        st.text_area("COPY EVERYTHING FROM HERE:", raw_text, height=600)
+                        
+                        # Show length info
+                        st.info(f"Response length: {len(raw_text)} characters")
+                        
+                        # 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")
                         
-                        # Try to extract JSON
-                        st.subheader("Extracted JSON Attempt:")
-                        extracted_json = extract_test_json_test(raw_text)
-                        if extracted_json:
-                            st.success("✅ JSON successfully extracted!")
-                            st.json(extracted_json)
-                        else:
-                            st.error("❌ Failed to extract JSON")
-                            st.text("Cleaned text for JSON extraction:")
-                            st.text(clean_json_text(raw_text))
-                            
                     except Exception as e:
                         st.error(f"Error: {str(e)}")
 
-# Test version of extract_and_fix_json
-def extract_and_fix_json_test(text, subject="General"):
-    """
-    Extract JSON from text and fix common issues - TEST VERSION
-    """
-    if not text or not isinstance(text, str):
-        return None
-    
-    # Clean the text
-    text = text.strip()
-    
-    # Remove markdown code blocks
-    text = re.sub(r'```json\s*', '', text)
-    text = re.sub(r'```\s*', '', 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:
-        return None
-    
-    json_str = text[start_idx:end_idx+1]
-    
-    try:
-        return json.loads(json_str)
-    except json.JSONDecodeError as e:
-        st.error(f"Initial JSON parse error: {str(e)}")
-        
-        # Fix common JSON issues
-        # 1. Fix unquoted keys
-        json_str = re.sub(r'([{,])\s*([a-zA-Z_][a-zA-Z0-9_]*)\s*:', r'\1"\2":', json_str)
-        
-        # 2. Fix trailing commas
-        json_str = re.sub(r',\s*}', '}', json_str)
-        json_str = re.sub(r',\s*]', ']', json_str)
-        
-        # 3. Fix single quotes to double quotes
-        json_str = re.sub(r"'([^']*)'", r'"\1"', json_str)
-        
-        # 4. Fix missing commas between objects in arrays
-        json_str = re.sub(r'}\s*{', '},{', json_str)
-        
-        try:
-            return json.loads(json_str)
-        except json.JSONDecodeError as e2:
-            st.error(f"Second JSON parse error: {str(e2)}")
-            
-            # Try to fix unescaped quotes
-            json_str = json_str.replace('\\"', '"')
-            
-            # Remove any non-ASCII characters if needed
-            json_str = json_str.encode('ascii', 'ignore').decode('ascii')
-            
-            try:
-                return json.loads(json_str)
-            except:
-                return None
-
-# Test version for test questions
-def extract_test_json_test(text):
-    """
-    Extract JSON from test response text
-    """
-    if not text or not isinstance(text, str):
-        return None
-    
-    text = text.strip()
-    
-    # Remove markdown code blocks
-    text = re.sub(r'```json\s*', '', text)
-    text = re.sub(r'```\s*', '', text)
-    text = re.sub(r'```javascript\s*', '', text)
-    text = re.sub(r'```python\s*', '', text)
-    
-    # Find the JSON part
-    start_idx = text.find('{')
-    end_idx = text.rfind('}')
-    
-    if start_idx == -1 or end_idx == -1 or end_idx <= start_idx:
-        # Maybe it's an array
-        start_idx = text.find('[')
-        end_idx = text.rfind(']')
-        
-        if start_idx == -1 or end_idx == -1 or end_idx <= start_idx:
-            return None
-    
-    json_str = text[start_idx:end_idx+1]
-    
-    try:
-        return json.loads(json_str)
-    except json.JSONDecodeError as e:
-        st.error(f"JSON parse error: {str(e)}")
-        
-        # Common fixes
-        # Fix 1: Unquoted keys
-        json_str = re.sub(r'([{,])\s*([a-zA-Z_][a-zA-Z0-9_]*)\s*:', r'\1"\2":', json_str)
-        
-        # Fix 2: Single quotes to double quotes
-        json_str = re.sub(r"'([^']*)'", r'"\1"', json_str)
-        
-        # Fix 3: Escape quotes in strings
-        json_str = json_str.replace('\\"', '\\\\"')
-        
-        # Fix 4: Remove trailing commas
-        json_str = re.sub(r',\s*}', '}', json_str)
-        json_str = re.sub(r',\s*]', ']', json_str)
-        
-        try:
-            return json.loads(json_str)
-        except:
-            return None
-
-def clean_json_text(text):
-    """
-    Clean text for JSON extraction
-    """
-    text = text.strip()
-    text = re.sub(r'```json\s*', '', text)
-    text = re.sub(r'```\s*', '', text)
-    text = re.sub(r'```javascript\s*', '', text)
-    text = re.sub(r'```python\s*', '', text)
-    return text
-
 # Run the test
 test_ai_responses()