Update app.py

app.py

@@ -1,11 +1,14 @@
-from gradio import Interface, File, Dropdown, Button, HTML
+import gradio as gr
 import pandas as pd
 import os
 from utils import pdf_to_text, align_text
 
 def process_files(source_file, target_file, lang1, lang2):
     if source_file is None or target_file is None:
-        return "Please upload both PDF files."
+        return "Please upload both PDF files.", None
+
+    if lang1 == lang2:
+        return "Please select different languages.", None
 
     # Convert PDFs to text
     text_content1 = pdf_to_text(source_file.name)
@@ -24,22 +27,21 @@ def process_files(source_file, target_file, lang1, lang2):
     return aligned_html, excel_path
 
 # Define the Gradio interface
-interface = Interface(
-    fn=process_files,
-    inputs=[
-        File(label="Upload Source PDF"),
-        File(label="Upload Target PDF"),
-        Dropdown(choices=["en", "es", "fr", "ch", "ar", "ru", "pt", "sw"], label="Select Language 1"),
-        Dropdown(choices=["en", "es", "fr", "ch", "ar", "ru", "pt", "sw"], label="Select Language 2"),
-    ],
-    outputs=[
-        HTML(label="Aligned DataFrame"),
-        File(label="Download Aligned Data as Excel")
-    ],
-    title="PDF Text Aligner",
-    live=True,
-    description="Upload two PDF files and select languages to align the text."
-)
+with gr.Blocks() as interface:
+    gr.Markdown("# PDF Text Aligner\nUpload two PDF files and select languages to align the text.")
+    source_file = gr.File(label="Upload Source PDF")
+    target_file = gr.File(label="Upload Target PDF")
+    lang1 = gr.Dropdown(choices=["en", "es", "fr", "ch", "ar", "ru", "pt-br", "sw"], label="Select Language 1")
+    lang2 = gr.Dropdown(choices=["en", "es", "fr", "ch", "ar", "ru", "pt-br", "sw"], label="Select Language 2", value="es")
+    start_button = gr.Button(value="Start")
+    aligned_html = gr.HTML(label="Aligned DataFrame")
+    download_button = gr.File(label="Download Aligned Data as Excel")
+
+    start_button.click(
+        fn=process_files,
+        inputs=[source_file, target_file, lang1, lang2],
+        outputs=[aligned_html, download_button]
+    )
 
 if __name__ == "__main__":
     interface.launch()