Update app.py

app.py

@@ -3082,7 +3082,10 @@ def show_story_stitching_options():
 def generate_stitched_story(story: List[dict], style: str, detail_level: str, theme: str, level: str) -> Dict[str, str]:
     """Generate a polished version of the story based on selected style and detail level"""
     try:
-        # Convert detail level to English
+        logging.info("Starting story generation...")
+        logging.info(f"Input params - Style: {style}, Detail: {detail_level}, Theme: {theme}, Level: {level}")
+        
+        # แปลง detail level เป็นภาษาอังกฤษ
         detail_map = {
             "น้อย": "minimal",
             "ปานกลาง": "moderate",
@@ -3090,178 +3093,100 @@ def generate_stitched_story(story: List[dict], style: str, detail_level: str, th
         }
         detail_level_en = detail_map.get(detail_level, "moderate")
         
-        # Extract style name
+        # ดึงชื่อสไตล์
         style_name = style.split(" - ")[0]
         
-        # Create style-specific instructions
-        style_instructions = {
-            "Classic Fairytale": """
-                Create a classic fairytale style with:
-                - Traditional opening and closing phrases
-                - Magical and whimsical elements
-                - Clear moral or lesson
-                - Flowing, storybook language
-            """,
-            "Modern Adventure": """
-                Create a contemporary adventure style with:
-                - Dynamic pacing
-                - Modern settings and references
-                - Engaging action sequences
-                - Relatable characters
-            """,
-            "Poetic Style": """
-                Create a poetic narrative style with:
-                - Rich imagery and metaphors
-                - Rhythmic language
-                - Evocative descriptions
-                - Artistic expression
-            """,
-            "Simple and Clear": """
-                Create a clear, straightforward style with:
-                - Direct language
-                - Clear sequence of events
-                - Focused storytelling
-                - Easy-to-follow structure
-            """
-        }
-        
-        # Create detail level instructions
-        detail_instructions = {
-            "minimal": "Keep descriptions concise and focus on key events.",
-            "moderate": "Balance description and action with moderate detail.",
-            "detailed": "Include rich descriptions and elaborate on scenes and characters."
-        }
-        
-        # Generate polished version
-        response = client.chat.completions.create(
-            model="gpt-4o-mini",
-            messages=[
-                {
-                    "role": "system",
-                    "content": f"""
-                    You are a professional children's story editor and translator.
-                    
-                    Style Guidelines:
-                    {style_instructions[style_name]}
-                    
-                    Detail Level:
-                    {detail_instructions[detail_level_en]}
-                    
-                    Educational Level: {level}
-                    
-                    Create a polished version that:
-                    1. Maintains the original story's core elements
-                    2. Adapts to the requested style
-                    3. Provides appropriate detail level
-                    4. Ensures age-appropriate language
-                    5. Creates a cohesive narrative flow
-                    """
-                },
-                {
-                    "role": "user",
-                    "content": f"Original story:\n{generate_story_summary(story)}\n\nCreate a polished version:"
-                }
-            ],
-            max_tokens=1500,
-            temperature=0.7
-        )
-        
-        polished_english = response.choices[0].message.content.strip()
-        
-        # Generate Thai translation with matching style
-        translation_response = client.chat.completions.create(
-            model="gpt-4o-mini",
-            messages=[
-                {
-                    "role": "system",
-                    "content": f"""
-                    You are a professional Thai translator specializing in children's literature.
-                    
-                    Create a Thai translation that:
-                    1. Matches the {style_name} style
-                    2. Maintains the same detail level ({detail_level_en})
-                    3. Uses natural, flowing Thai language
-                    4. Preserves the story's tone and feeling
-                    5. Adapts cultural elements appropriately
-                    """
-                },
-                {
-                    "role": "user",
-                    "content": f"Translate this story to Thai:\n{polished_english}"
-                }
-            ],
-            max_tokens=1500,
-            temperature=0.7
-        )
-        
-        thai_translation = translation_response.choices[0].message.content.strip()
+        # สร้าง prompt
+        story_summary = generate_story_summary(story)
+        logging.info(f"Story summary length: {len(story_summary)}")
         
-        return {
-            'original': generate_story_summary(story),
+        # Generate English version
+        try:
+            logging.info("Generating English version...")
+            response = client.chat.completions.create(
+                model="gpt-4",
+                messages=[
+                    {
+                        "role": "system",
+                        "content": f"You are a children's story editor. Create a {style_name} style story with {detail_level_en} detail level."
+                    },
+                    {
+                        "role": "user",
+                        "content": f"Original story:\n{story_summary}\n\nCreate a polished version:"
+                    }
+                ],
+                max_tokens=1000,
+                temperature=0.7
+            )
+            polished_english = response.choices[0].message.content.strip()
+            logging.info("English version generated successfully")
+        except Exception as e:
+            logging.error(f"Error generating English version: {str(e)}")
+            raise
+
+        # Generate Thai translation
+        try:
+            logging.info("Generating Thai translation...")
+            translation_response = client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=[
+                    {
+                        "role": "system",
+                        "content": "You are a Thai translator. Create a natural Thai translation."
+                    },
+                    {
+                        "role": "user",
+                        "content": f"Translate this story to Thai:\n{polished_english}"
+                    }
+                ],
+                max_tokens=1000,
+                temperature=0.7
+            )
+            thai_translation = translation_response.choices[0].message.content.strip()
+            logging.info("Thai translation generated successfully")
+        except Exception as e:
+            logging.error(f"Error generating Thai translation: {str(e)}")
+            raise
+
+        result = {
+            'original': story_summary,
             'polished_english': polished_english,
             'thai_translation': thai_translation,
             'style': style_name,
             'detail_level': detail_level_en
         }
         
+        logging.info("Story generation completed successfully")
+        return result
+        
     except Exception as e:
-        logging.error(f"Error generating stitched story: {str(e)}")
+        logging.error(f"Error in story generation: {str(e)}")
         raise
 
 def show_stitched_result(story_data: Dict[str, str]):
     """Display the stitched story result with enhanced formatting"""
     try:
-        st.markdown("""
-            <div style="
-                background-color: #f8f9fa;
-                padding: 20px;
-                border-radius: 10px;
-                margin: 20px 0;
-            ">
-                <h2 style="color: #1565c0; text-align: center; margin-bottom: 20px;">
-                    ✨ เรื่องราวฉบับสมบูรณ์ | Complete Story
-                </h2>
-                
-                <div style="
-                    background-color: white;
-                    padding: 20px;
-                    border-radius: 8px;
-                    margin-bottom: 20px;
-                    border: 1px solid #e0e0e0;
-                ">
-                    <h3 style="color: #2e7d32; margin-bottom: 10px;">
-                        English Version
-                    </h3>
-                    <div style="
-                        color: #333;
-                        font-size: 1.1em;
-                        line-height: 1.6;
-                        white-space: pre-line;
-                    ">
-                        {story_data['polished_english']}
-                    </div>
-                </div>
-                
-                <div style="
-                    background-color: white;
-                    padding: 20px;
-                    border-radius: 8px;
-                    border: 1px solid #e0e0e0;
-                ">
-                    <h3 style="color: #2e7d32; margin-bottom: 10px;">
-                        ฉบับภาษาไทย
-                    </h3>
-                    <div style="
-                        color: #333;
-                        font-size: 1.1em;
-                        line-height: 1.6;
-                        white-space: pre-line;
-                    ">
-                        {story_data['thai_translation']}
-                    </div>
-                </div>
-            </div>
-        """, unsafe_allow_html=True)
+        st.markdown("## 📖 Your Polished Story | เรื่องราวฉบับสมบูรณ์")
+        
+        # English Version
+        st.markdown("### 🇬🇧 English Version")
+        st.markdown(
+            f"""<div style="background-color: #f8f9fa; padding: 20px; 
+            border-radius: 10px; margin: 10px 0; border-left: 4px solid #1565c0;">
+            {story_data['polished_english']}
+            </div>""", 
+            unsafe_allow_html=True
+        )
+        
+        # Thai Version
+        st.markdown("### 🇹🇭 ฉบับภาษาไทย")
+        st.markdown(
+            f"""<div style="background-color: #f8f9fa; padding: 20px; 
+            border-radius: 10px; margin: 10px 0; border-left: 4px solid #4caf50;">
+            {story_data['thai_translation']}
+            </div>""", 
+            unsafe_allow_html=True
+        )
         
         # Save options
         st.markdown("### 💾 บันทึกเรื่องราว")
@@ -3282,6 +3207,7 @@ def show_stitched_result(story_data: Dict[str, str]):
                         mime="application/pdf",
                         key="download_final_pdf"
                     )
+                    st.success("สร้าง PDF เรียบร้อยแล้ว!")
                 except Exception as e:
                     logging.error(f"Error saving PDF: {str(e)}")
                     st.error("เกิดข้อผิดพลาดในการสร้างไฟล์ PDF")
@@ -3300,6 +3226,7 @@ def show_stitched_result(story_data: Dict[str, str]):
                         mime="application/json",
                         key="download_final_text"
                     )
+                    st.success("บันทึกข้อความเรียบร้อยแล้ว!")
                 except Exception as e:
                     logging.error(f"Error saving text: {str(e)}")
                     st.error("เกิดข้อผิดพลาดในการบันทึกข้อความ")