Update app.py

app.py

@@ -1,121 +1,88 @@
+from transformers import pipeline
+from fpdf import FPDF
 import gradio as gr
 import pandas as pd
-from fpdf import FPDF
 from datetime import datetime
-import tempfile  # Use for temporary file handling
-import os
+import tempfile
+
+# Load text generation model
+generator = pipeline("text-generation", model="gpt-2")
 
-# Initialize an empty DataFrame to store all records of invoices (acting as our ledger)
+# Initialize an empty DataFrame to store all records of invoices
 ledger = pd.DataFrame(columns=["Head", "Company Name", "SP/Station", "Invoice No", "Date", "Amount"])
 
-# Function to add a new invoice under a specific head
+# Function to add a new invoice
 def add_invoice(head, company_name, sp_station, invoice_no, date, amount):
     global ledger
-    try:
-        # Add the new record to the ledger
-        new_row = {
-            "Head": head,
-            "Company Name": company_name,
-            "SP/Station": sp_station,
-            "Invoice No": invoice_no,
-            "Date": date,
-            "Amount": float(amount)
-        }
-        ledger = pd.concat([ledger, pd.DataFrame([new_row])], ignore_index=True)
-
-        # Filter the ledger for the selected head
-        filtered_ledger = ledger[ledger["Head"] == head]
-
-        # Calculate the total for the selected head
-        total_amount = filtered_ledger["Amount"].sum()
-        return filtered_ledger, f"Total for {head}: {total_amount:.2f}"
-
-    except Exception as e:
-        print(f"Error during adding invoice: {e}")
-        return pd.DataFrame(), "Error during adding invoice"
-
-# Function to search and display all previous records by head
+    new_row = {
+        "Head": head,
+        "Company Name": company_name,
+        "SP/Station": sp_station,
+        "Invoice No": invoice_no,
+        "Date": date,
+        "Amount": float(amount)
+    }
+    ledger = pd.concat([ledger, pd.DataFrame([new_row])], ignore_index=True)
+    filtered_ledger = ledger[ledger["Head"] == head]
+    total_amount = filtered_ledger["Amount"].sum()
+    return filtered_ledger, f"Total for {head}: {total_amount:.2f}"
+
+# Function to search and display all records by head
 def search_by_head(head):
-    global ledger
-    try:
-        # Filter the ledger for the selected head
-        filtered_ledger = ledger[ledger["Head"] == head]
-
-        # Calculate the total for the selected head
-        total_amount = filtered_ledger["Amount"].sum()
-
-        return filtered_ledger, f"Total for {head}: {total_amount:.2f}"
+    filtered_ledger = ledger[ledger["Head"] == head]
+    total_amount = filtered_ledger["Amount"].sum()
+    return filtered_ledger, f"Total for {head}: {total_amount:.2f}"
 
-    except Exception as e:
-        print(f"Error during search by head: {e}")
-        return pd.DataFrame(), "Error during search by head"
-
-# Function to generate a PDF report for a specific head
+# Function to generate a PDF report
 def generate_ledger_pdf(head):
     global ledger
-    try:
-        # Ensure the head is not empty
-        if head.strip() == "":
-            return None
-
-        # Use a temporary directory to save the file
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
-            pdf_file_path = tmp_file.name  # Get the temporary file path
-
-            # Create a PDF
-            pdf = FPDF()
-            pdf.add_page()
-            
-            # Title and header
-            pdf.set_font("Arial", 'B', 16)
-            pdf.cell(0, 10, f"Ledger Report for {head}", ln=True, align="C")
-            pdf.set_font("Arial", 'I', 12)
-            current_date = datetime.now().strftime("%d/%m/%Y")
-            pdf.cell(0, 10, f"Report Date: {current_date}", ln=True, align="C")
+
+    # Generate some content for the PDF using the Hugging Face model
+    generated_text = generator(f"Generate a report summary for {head}.", max_length=50)[0]['generated_text']
+
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
+        pdf_file_path = tmp_file.name
+
+        pdf = FPDF()
+        pdf.add_page()
+        
+        # Title and generated content
+        pdf.set_font("Arial", 'B', 16)
+        pdf.cell(0, 10, f"Ledger Report for {head}", ln=True, align="C")
+        pdf.set_font("Arial", 'I', 12)
+        pdf.cell(0, 10, f"Generated Content: {generated_text}", ln=True, align="C")
+        pdf.ln(10)
+
+        # Table headers
+        pdf.set_font("Arial", 'B', 12)
+        pdf.cell(50, 10, "Company Name", border=1, align="C")
+        pdf.cell(40, 10, "SP/Station", border=1, align="C")
+        pdf.cell(30, 10, "Invoice No", border=1, align="C")
+        pdf.cell(30, 10, "Date", border=1, align="C")
+        pdf.cell(40, 10, "Amount", border=1, ln=True, align="C")
+
+        filtered_ledger = ledger[ledger["Head"] == head]
+        if not filtered_ledger.empty:
+            pdf.set_font("Arial", size=12)
+            for _, row in filtered_ledger.iterrows():
+                pdf.cell(50, 10, str(row['Company Name']), border=1, align="C")
+                pdf.cell(40, 10, str(row['SP/Station']), border=1, align="C")
+                pdf.cell(30, 10, str(row['Invoice No']), border=1, align="C")
+                pdf.cell(30, 10, str(row['Date']), border=1, align="C")
+                pdf.cell(40, 10, f"{row['Amount']:.2f}", border=1, ln=True, align="C")
+
+            # Total amount for the head
+            total_amount = filtered_ledger["Amount"].sum()
             pdf.ln(10)
-            
-            # Table headers
             pdf.set_font("Arial", 'B', 12)
-            pdf.cell(50, 10, "Company Name", border=1, align="C")
-            pdf.cell(40, 10, "SP/Station", border=1, align="C")
-            pdf.cell(30, 10, "Invoice No", border=1, align="C")
-            pdf.cell(30, 10, "Date", border=1, align="C")
-            pdf.cell(40, 10, "Amount", border=1, ln=True, align="C")
-
-            # Filter records by head
-            filtered_ledger = ledger[ledger["Head"] == head]
-            if filtered_ledger.empty:
-                pdf.cell(0, 10, f"No records found for {head}", ln=True, align="C")
-            else:
-                pdf.set_font("Arial", size=12)
-                
-                for _, row in filtered_ledger.iterrows():
-                    pdf.cell(50, 10, str(row['Company Name']), border=1, align="C")
-                    pdf.cell(40, 10, str(row['SP/Station']), border=1, align="C")
-                    pdf.cell(30, 10, str(row['Invoice No']), border=1, align="C")
-                    pdf.cell(30, 10, str(row['Date']), border=1, align="C")
-                    pdf.cell(40, 10, f"{row['Amount']:.2f}", border=1, ln=True, align="C")
-
-                # Total amount for the head
-                total_amount = filtered_ledger["Amount"].sum()
-                pdf.ln(10)  # Add some space before the total
-                pdf.set_font("Arial", 'B', 12)
-                pdf.cell(150, 10, "Total Amount", border=1, align="R")
-                pdf.cell(40, 10, f"{total_amount:.2f}", border=1, ln=True, align="C")
-
-            # Save PDF file
-            pdf.output(pdf_file_path)
-
-            # Return the file path for download
-            return pdf_file_path
-
-    except Exception as e:
-        print(f"Error during PDF generation: {e}")
-        return None
+            pdf.cell(150, 10, "Total Amount", border=1, align="R")
+            pdf.cell(40, 10, f"{total_amount:.2f}", border=1, ln=True, align="C")
+
+        pdf.output(pdf_file_path)
+        return pdf_file_path
 
 # Gradio interface
 with gr.Blocks() as app:
-    # Section for adding new invoices with Head and Company Name fields
     with gr.Row():
         head = gr.Textbox(label="Head")
         company_name = gr.Textbox(label="Company Name")
@@ -124,36 +91,19 @@ with gr.Blocks() as app:
         date = gr.Textbox(label="Date")
         amount = gr.Number(label="Amount")
         add_button = gr.Button("Add Invoice")
-    
-    # Section for searching and displaying ledger by head
+
     with gr.Row():
         search_head = gr.Textbox(label="Search by Head")
         search_button = gr.Button("Search Ledger")
         ledger_records = gr.DataFrame(label="Ledger Records")
         total_display = gr.Textbox(label="Total Amount", interactive=False)
-    
-    # Section for generating and displaying PDF link
+
     with gr.Row():
         generate_pdf_button = gr.Button("Generate PDF")
         pdf_output = gr.File(label="Download PDF")
 
-    # Define button functionalities
     add_button.click(add_invoice, [head, company_name, sp_station, invoice_no, date, amount], [ledger_records, total_display])
-    
-    def search_ledger(head):
-        filtered_ledger, total = search_by_head(head)
-        return filtered_ledger, total
-
-    search_button.click(search_ledger, [search_head], [ledger_records, total_display])
-
-    def generate_pdf(head):
-        pdf_path = generate_ledger_pdf(head)
-        if pdf_path:
-            return pdf_path
-        else:
-            return None
-
-    generate_pdf_button.click(generate_pdf, [search_head], [pdf_output])
+    search_button.click(search_by_head, [search_head], [ledger_records, total_display])
+    generate_pdf_button.click(generate_ledger_pdf, [search_head], [pdf_output])
 
-# Launch the app
 app.launch(share=True)