Update app.py

app.py

@@ -145,6 +145,45 @@ def process_image1(image, use_llm, use_context):
 
 
 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']
+        ascii_braille = state['ascii_braille']
+        
+        # If ASCII version is not available, use the Unicode version
+        if not ascii_braille:
+            ascii_braille = state['braille_text']
+    except:
+        # Fallback for backward compatibility
+        if isinstance(state, tuple) and len(state) == 2:
+            original_text, braille_text = state
+            ascii_braille = braille_text
+        else:
+            return None
+    
+    comparison = (pdf_type == "Side-by-Side Comparison")
+    
+    try:
+        pdf_buffer = generate_pdf(original_text, ascii_braille, 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_pdf11(state, pdf_title, pdf_type):
     """Create a PDF file for download."""
     if state is None or len(state) != 2:
         return None