Update app.py

app.py

@@ -85,11 +85,43 @@ def process_image(image, use_llm, use_context):
     except Exception as e:
         return f"Error processing image: {str(e)}", "", "", None
 
+
 def create_pdf(state, pdf_title, pdf_type):
     """Create a PDF file for download."""
     if state is None or len(state) != 2:
         return None
     
+    original_text, braille_text = state
+    
+    # Get ASCII representation for PDF
+    try:
+        braille_result = text_to_braille(original_text, use_context=False)
+        ascii_braille = braille_result.get('formatted_ascii', braille_text)
+    except:
+        ascii_braille = braille_text
+    
+    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_pdf2(state, pdf_title, pdf_type):
+    """Create a PDF file for download."""
+    if state is None or len(state) != 2:
+        return None
+    
     original_text, braille_text = state
     comparison = (pdf_type == "Side-by-Side Comparison")