Update app.py

app.py

@@ -1,17 +1,25 @@
 import gradio as gr
-from parse_bhel_po import extract_po_details  # Import the extraction function from parse_bhel_po.py
+from parse_bhel_po import extract_po_details  # Import directly from parse_bhel_po.py
 
-def process_uploaded_pdf(uploaded_file):
-    # Pass the file-like object directly to extract_po_details
-    df = extract_po_details(uploaded_file)
-    return df
+# Define a function to handle file processing based on the selected format
+def process_pdf(uploaded_file, selected_format):
+    if selected_format == "BHEL":
+        # Use the BHEL parser to extract data
+        df = extract_po_details(uploaded_file)
+        return df.to_string(index=False)
+    else:
+        return "Selected format not supported."
 
-# Create a Gradio interface for the application
+# Set up Gradio interface
 interface = gr.Interface(
-    fn=process_uploaded_pdf,
-    inputs="file",               # "file" input type to handle file uploads
-    outputs="dataframe",         # Display the extracted data as a dataframe
-    title="BHEL PO Details Extraction"
+    fn=process_pdf,
+    inputs=[
+        gr.Dropdown(label="Select Format", choices=["BHEL"], value="BHEL"),
+        gr.File(label="Upload PDF"),
+    ],
+    outputs="text",
+    title="PO Details Extraction",
+    description="Select the format and upload the PO PDF to extract details."
 )
 
 if __name__ == "__main__":