Update app.py

app.py

@@ -182,8 +182,40 @@ def create_pdf_v2(state, pdf_title, pdf_type):
         print(f"Error generating PDF: {str(e)}")
         return None
 
-
 def create_pdf(state, pdf_title, pdf_type):
+    """Create a PDF file for download."""
+    if state is None:
+        return None
+    
+    # Extract data from state
+    try:
+        original_text = state['original_text']
+        braille_text = state['braille_text']  # Use Unicode Braille text
+    except:
+        # Fallback for backward compatibility
+        if isinstance(state, tuple) and len(state) == 2:
+            original_text, braille_text = state
+        else:
+            return None
+    
+    comparison = (pdf_type == "Side-by-Side Comparison")
+    
+    try:
+        pdf_buffer = generate_pdf(original_text, braille_text, pdf_title, comparison)
+        
+        # Create a temporary file to save the PDF
+        temp_file_path = f"/tmp/{pdf_title.replace(' ', '_').lower()}.pdf"
+        
+        # Write the buffer to a file
+        with open(temp_file_path, "wb") as f:
+            f.write(pdf_buffer.getvalue())
+        
+        return temp_file_path
+    except Exception as e:
+        print(f"Error generating PDF: {str(e)}")
+        return None
+
+def create_pdf_v1_working(state, pdf_title, pdf_type):
     """Create a PDF file for download."""
     if state is None or len(state) != 2:
         return None