Update app.py

app.py

@@ -39,8 +39,9 @@ def pdf_to_text(file):
         image_list = page.get_images(full=True)
         for img in image_list:
             xref, name, ext, color_space, width, height, bpc, image_data, image_mask, smask_data = img
-            # Decode image_data from base64 before opening it
-            image_data = base64.b64decode(image_data)
+            # Ensure image_data is a bytes object before opening it
+            if isinstance(image_data, str):
+                image_data = image_data.encode()  # Convert string to bytes if necessary
             image = Image.open(io.BytesIO(image_data))
             latex_code = image_to_latex(image)
             page_text += "\n" + latex_code  # Add LaTeX code to page text
@@ -56,12 +57,13 @@ def pdf_to_text(file):
     with open(output_file_name, 'w') as f:
         f.write(full_text)
 
-    return output_file_name, page_number
+        return f"{output_file_name},{page_number}"
 
 
 iface = gr.Interface(fn=pdf_to_text,
                      inputs=gr.inputs.File(label="Your PDF"),
-                     outputs=gr.outputs.File(label="Download TXT"),
+                     outputs="text",
                      title="PDF to TXT",
                      description="Convert your PDF files to clean text")
 iface.launch()
+