Update app.py

app.py

@@ -16,54 +16,46 @@ def convert(file, out_format):
         # Convert the file
         doc = converter.convert(file.name)
 
-        # Get the base filename without extension
-        base_filename = os.path.splitext(os.path.basename(file.name))[0]
-
         # Export based on the selected format
         if out_format == "Markdown":
             # Save as Markdown file
-            output_filename = f"{base_filename}.md"
             with tempfile.NamedTemporaryFile(delete=False, suffix=".md") as temp_file:
                 temp_filename = temp_file.name
                 with open(temp_filename, 'w') as f:
                     f.write(doc.document.export_to_markdown())
-                return temp_filename, output_filename  # Return file for download
+                return temp_filename  # Return file for download
 
         elif out_format == "HTML":
             # Save as HTML file
-            output_filename = f"{base_filename}.html"
             with tempfile.NamedTemporaryFile(delete=False, suffix=".html") as temp_file:
                 temp_filename = temp_file.name
                 with open(temp_filename, 'w') as f:
                     f.write(doc.document.export_to_html())
-                return temp_filename, output_filename  # Return file for download
+                return temp_filename  # Return file for download
 
         elif out_format == "JSON":
             # Save as JSON file (lossless serialization)
-            output_filename = f"{base_filename}.json"
             doctags = doc.document.export_to_dict()  # Correct method for JSON export
             with tempfile.NamedTemporaryFile(delete=False, suffix=".json") as temp_file:
                 temp_filename = temp_file.name
                 with open(temp_filename, 'w') as f:
                     json.dump(doctags, f, indent=4)
-                return temp_filename, output_filename  # Return file for download
+                return temp_filename  # Return file for download
 
         elif out_format == "Text":
             # Save as Text file (plain text)
-            output_filename = f"{base_filename}.txt"
             with tempfile.NamedTemporaryFile(delete=False, suffix=".txt") as temp_file:
                 temp_filename = temp_file.name
                 with open(temp_filename, 'w') as f:
                     f.write(doc.document.export_to_text())  # Correct method for plain text extraction
-                return temp_filename, output_filename  # Return file for download
+                return temp_filename  # Return file for download
 
         elif out_format == "Doctags":
             # Save as Doctags file
-            output_filename = f"{base_filename}.doctags"
             with tempfile.NamedTemporaryFile(delete=False, suffix=".doctags") as temp_file:
                 temp_filename = temp_file.name
                 doc.document.save_as_doctags(temp_filename)
-                return temp_filename, output_filename  # Return file for download
+                return temp_filename  # Return file for download
 
         else:
             return "Unsupported output format"
@@ -114,12 +106,8 @@ with gr.Blocks() as demo:
     out = gr.File(label="📄 Download Output File")
 
     # Wire it up
-    def on_convert(file, fmt):
-        output_file, output_filename = convert(file, fmt)
-        return output_file  # Return file path for download
-
-    btn.click(on_convert, inputs=[inp, fmt], outputs=out)
+    btn.click(fn=convert, inputs=[inp, fmt], outputs=out)
 
 # 4) Launch
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch().