Update federal_electric.py

federal_electric.py

@@ -89,31 +89,11 @@ def extract_po_data(pdf_file):
 
     return df
 
-def process_and_save(pdf_file, output_format):
+def process_pdf(file):
     """
-    Processes the uploaded PDF and saves the extracted data as an Excel or CSV file.
+    Processes the uploaded PDF and saves the extracted data.
     """
-    df = extract_po_data(pdf_file.name)
-    
-    # Save the file in the desired format
-    output_file = f"output.{output_format}"
-    if output_format == "csv":
-        df.to_csv(output_file, index=False)
-    elif output_format == "xlsx":
-        df.to_excel(output_file, index=False, engine="openpyxl")
-    
-    return output_file
-
-# Gradio interface function
-def gradio_interface(pdf_file, output_format):
-    output_file = process_and_save(pdf_file, output_format)
-    return output_file
-
-def process_pdf(file):
     try:
-        # Extract text from the PDF
-        text = extract_text_from_pdf(file)
-        
         # Process the extracted text into a DataFrame
         df = extract_po_data(file.name)
         
@@ -124,16 +104,16 @@ def process_pdf(file):
     except Exception as e:
         return None, f"Error during processing: {str(e)}"
 
-
-
-
 # Gradio app interface
 iface = gr.Interface(
-    fn=gradio_interface,
-    inputs=[gr.File(label="Upload PDF"), gr.Radio(["csv", "xlsx"], label="Output Format")],
-    outputs=gr.File(label="Download Output"),
+    fn=process_pdf,
+    inputs=[gr.File(label="Upload PDF")],
+    outputs=[
+        gr.File(label="Download Extracted Data"),
+        gr.Textbox(label="Status")
+    ],
     title="Enhanced PO Data Extractor",
-    description="Extract data from Purchase Orders, including multi-line descriptions, and clean unwanted text or symbols. Download as CSV or Excel."
+    description="Extract data from Purchase Orders, including multi-line descriptions, and clean unwanted text or symbols."
 )
 
 if __name__ == "__main__":