Update app.py

app.py

@@ -1809,14 +1809,41 @@ def handle_story_submission(text: str):
         update_achievements()
         
         # Generate AI continuation
-        if not feedback_data.get('has_errors'):
-            with st.spinner("กำลังต่อเรื่อง..."):
-                ai_response = generate_story_continuation(text, st.session_state.level)
+        try:
+            logging.info("Attempting to generate AI continuation...")
+            # ใช้ข้อความที่ถูกต้องสำหรับการต่อเรื่อง
+            text_for_continuation = feedback_data['corrected'] if feedback_data.get('has_errors') else text
+            
+            ai_response = generate_story_continuation(text_for_continuation, st.session_state.level)
+            
+            # Log the AI response
+            logging.info(f"AI Response generated: {ai_response}")
+            
+            if ai_response and ai_response.strip():
                 st.session_state.story.append({
                     "role": "AI",
                     "content": ai_response,
                     "timestamp": datetime.now().isoformat()
                 })
+                logging.info("AI response added to story successfully")
+            else:
+                logging.error("AI generated empty response")
+                # กรณีที่ AI ไม่สร้างประโยค ให้สร้างประโยคง่ายๆ ตาม theme
+                fallback_response = generate_fallback_response(st.session_state.current_theme, st.session_state.level)
+                st.session_state.story.append({
+                    "role": "AI",
+                    "content": fallback_response,
+                    "timestamp": datetime.now().isoformat()
+                })
+        except Exception as e:
+            logging.error(f"Error generating AI continuation: {str(e)}")
+            # สร้าง fallback response เมื่อเกิดข้อผิดพลาด
+            fallback_response = generate_fallback_response(st.session_state.current_theme, st.session_state.level)
+            st.session_state.story.append({
+                "role": "AI",
+                "content": fallback_response,
+                "timestamp": datetime.now().isoformat()
+            })
         
         # Update session stats
         update_session_stats()
@@ -1831,6 +1858,31 @@ def handle_story_submission(text: str):
         logging.error(f"Error in story submission: {str(e)}")
         raise
 
+def generate_fallback_response(theme_id: str, level: str) -> str:
+    """Generate a simple fallback response when AI continuation fails"""
+    try:
+        theme = story_themes.get(theme_id, {})
+        if theme:
+            # ใช้คำศัพท์จาก theme และ level ที่เลือก
+            vocab = theme.get('vocabulary', {}).get(level, [])
+            if vocab:
+                word = random.choice(vocab)
+                
+                # สร้างประโยคตาม level
+                if level == 'Beginner':
+                    return f"The story continues with {word}..."
+                elif level == 'Intermediate':
+                    return f"Something interesting happened with the {word}."
+                else:  # Advanced
+                    return f"The mystery deepens as we discover more about the {word}."
+        
+        # Default fallback if no theme-specific response can be generated
+        return "The story continues..."
+        
+    except Exception as e:
+        logging.error(f"Error generating fallback response: {str(e)}")
+        return "What happens next?"
+
 def show_main_interface():
     """Display main story interface"""
     col1, col2 = st.columns([3, 1])