Update app.py

app.py

@@ -2486,6 +2486,9 @@ def show_save_dialog():
                 except Exception as e:
                     logging.error(f"Error saving progress: {str(e)}")
                     st.error("เกิดข้อผิดพลาดในการบันทึกความก้าวหน้า")
+    except Exception as e:
+        logging.error(f"Error showing save dialog: {str(e)}")
+        st.error("เกิดข้อผิดพลาดในการแสดงตัวเลือกการบันทึก")
 
 def reset_story_with_confirmation():
     """Reset story with confirmation dialog"""
@@ -2501,6 +2504,55 @@ def reset_story_with_confirmation():
         logging.error(f"Error in reset confirmation: {str(e)}")
         st.error("เกิดข้อผิดพลาดในการรีเซ็ตเรื่อง")
 
+def create_story_pdf():
+    """Create PDF of the story"""
+    try:
+        buffer = io.BytesIO()
+        doc = SimpleDocTemplate(buffer, pagesize=A4)
+        story_elements = []
+        
+        # Add title
+        styles = getSampleStyleSheet()
+        title_style = ParagraphStyle(
+            'CustomTitle',
+            parent=styles['Title'],
+            fontSize=24,
+            spaceAfter=30
+        )
+        story_elements.append(Paragraph("My Story", title_style))
+        
+        # Add content
+        content_style = ParagraphStyle(
+            'CustomBody',
+            parent=styles['Normal'],
+            fontSize=12,
+            leading=16,
+            spaceAfter=12
+        )
+        
+        for entry in st.session_state.story:
+            if entry['role'] == 'AI':
+                color = colors.blue
+            else:
+                color = colors.black
+            
+            p = Paragraph(
+                f"<font color={color}>{entry['content']}</font>",
+                content_style
+            )
+            story_elements.append(p)
+        
+        # Build PDF
+        doc.build(story_elements)
+        pdf = buffer.getvalue()
+        buffer.close()
+        
+        return pdf
+        
+    except Exception as e:
+        logging.error(f"Error creating PDF: {str(e)}")
+        raise
+
 # === 6. MAIN APPLICATION LOGIC ===
 def main():
     try: