Update app.py

app.py

@@ -3181,83 +3181,35 @@ def generate_stitched_story(story: List[dict], style: str, detail_level: str, th
 def show_stitched_result(story_data: Dict[str, str]):
     """Display the stitched story result with enhanced formatting"""
     try:
-        st.markdown("## 📖 Your Polished Story | เรื่องราวฉบับสมบูรณ์")
+        # Check if story data exists
+        if not story_data or not isinstance(story_data, dict):
+            st.error("ไม่พบข้อมูลเรื่องราว กรุณาลองใหม่อีกครั้ง")
+            return
+
+        # Header
+        st.markdown("### 📖 Your Polished Story | เรื่องราวฉบับสมบูรณ์")
         
         # English Version
-        st.markdown("### 🇬🇧 English Version")
+        st.subheader("🇬🇧 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>""", 
+            f'<div style="background-color: white; padding: 15px; border-radius: 8px; margin-bottom: 15px;">'
+            f'{story_data.get("polished_english", "No content available")}'
+            f'</div>',
             unsafe_allow_html=True
         )
         
         # Thai Version
-        st.markdown("### 🇹🇭 ฉบับภาษาไทย")
+        st.subheader("🇹🇭 ฉบับภาษาไทย")
         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>""", 
+            f'<div style="background-color: white; padding: 15px; border-radius: 8px;">'
+            f'{story_data.get("thai_translation", "ไม่พบเนื้อหา")}'
+            f'</div>',
             unsafe_allow_html=True
         )
-        
-        # Save options
-        st.markdown("### 💾 บันทึกเรื่องราว")
-        
-        save_col1, save_col2 = st.columns(2)
-        
-        with save_col1:
-            if st.button("📥 บันทึกเป็น PDF", 
-                        key="save_final_pdf", 
-                        use_container_width=True):
-                try:
-                    pdf_data = create_bilingual_story_pdf(story_data)
-                    file_name = f"my_story_{datetime.now().strftime('%Y%m%d_%H%M')}.pdf"
-                    st.download_button(
-                        "ดาวน์โหลด PDF",
-                        data=pdf_data,
-                        file_name=file_name,
-                        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")
-        
-        with save_col2:
-            if st.button("📝 บันทึกข้อความ", 
-                        key="save_final_text", 
-                        use_container_width=True):
-                try:
-                    text_data = json.dumps(story_data, ensure_ascii=False, indent=2)
-                    file_name = f"my_story_{datetime.now().strftime('%Y%m%d_%H%M')}.json"
-                    st.download_button(
-                        "ดาวน์โหลดข้อความ",
-                        data=text_data,
-                        file_name=file_name,
-                        mime="application/json",
-                        key="download_final_text"
-                    )
-                    st.success("บันทึกข้อความเรียบร้อยแล้ว!")
-                except Exception as e:
-                    logging.error(f"Error saving text: {str(e)}")
-                    st.error("เกิดข้อผิดพลาดในการบันทึกข้อความ")
-        
-        # New story option
-        if st.button("🔄 เริ่มเรื่องใหม่", 
-                    key="new_story_after_stitch", 
-                    use_container_width=True):
-            if st.checkbox("✅ ยืนยันการเริ่มใหม่", 
-                         key="confirm_new_after_stitch"):
-                reset_story()
-                st.rerun()
-                
+
     except Exception as e:
         logging.error(f"Error showing stitched result: {str(e)}")
-        st.error("เกิดข้อผิดพลาดในการแสดงผลลัพธ์")
+        st.error("เกิดข้อผิดพลาดในการแสดงผลเรื���องราว กรุณาลองใหม่อีกครั้ง")
 
 def show_save_dialog():
     """Display save options dialog"""