Update app.py

app.py

@@ -1,45 +1,35 @@
 import gradio as gr
-
-# Placeholder function for PO extraction
-# Replace this with the actual logic for processing the selected format
-def process_format(format_name):
-    # Simulated logic for different formats
-    if format_name == "BHEL.py":
-        return "Processing in BHEL.py format..."
-    elif format_name == "Federal Electric.py":
-        return "Processing in Federal Electric.py format..."
-    elif format_name == "AL-NISF":
-        return "Processing in AL-NISF format..."
+import pandas as pd
+from BHEL import parse_bhel_pdf
+from FederalElectric import parse_federal_electric_pdf
+from ALNISF import parse_alnisf_pdf
+
+def process_pdf(file, format_type):
+    # Select the appropriate parser based on format type
+    if format_type == "BHEL.py":
+        df = parse_bhel_pdf(file.name)
+    elif format_type == "Federal Electric.py":
+        df = parse_federal_electric_pdf(file.name)
+    elif format_type == "AL-NISF":
+        df = parse_alnisf_pdf(file.name)
     else:
-        return "Unknown format selected."
-
-# Define the dropdown options
-format_options = ["BHEL.py", "Federal Electric.py", "AL-NISF"]
-
-# Build the Gradio app
-def app():
-    dropdown = gr.Dropdown(
-        choices=format_options,
-        label="Select Format",
-        value="BHEL.py",  # Default value
-        interactive=True,
-    )
-
-    output = gr.Textbox(label="Result", interactive=False)
+        return "Unsupported format selected", None
 
-    # Update function to display result based on selected format
-    def update(format_name):
-        return process_format(format_name)
+    # Save the DataFrame to an Excel file
+    output_file = f"{format_type}_Data.xlsx"
+    df.to_excel(output_file, index=False)
 
-    interface = gr.Interface(
-        fn=update,
-        inputs=dropdown,
-        outputs=output,
-        title="PO Extraction Format Selector",
-        description="Select the desired format from the dropdown, and the app will process accordingly."
-    )
+    return output_file
 
-    return interface
+# Gradio Interface
+iface = gr.Interface(
+    fn=process_pdf,
+    inputs=[
+        gr.File(label="Upload PDF"),
+        gr.Dropdown(choices=["BHEL.py", "Federal Electric.py", "AL-NISF"], label="Select Format")
+    ],
+    outputs=gr.File(label="Download Excel")
+)
 
 if __name__ == "__main__":
-    app().launch()
+    iface.launch()