Update app.py

app.py

@@ -1231,7 +1231,7 @@ def handle_large_dataset(df, create_document,isDataFrame):
             #Logic to generate pdfs with employee name and account number
             for index, row in df.iterrows():
                 # Create a PDF for each row
-                create_pdf(row['Account Name'], row['Account ID'])
+                create_pdf(row['COMPANY'], row['EMPLOYEE NAME'])
             create_document = False
             docstatus += f" {total_rows} documents are created successfully."
         print(sample_table)
@@ -1244,18 +1244,17 @@ def handle_large_dataset(df, create_document,isDataFrame):
 
     else:
         return "Your query returns a large dataset which is not supported in the current version. Please try a different query."
-def create_pdf(name,id):
+def create_pdf(cname,ename):
    
 
-    filled = FormWrapper("Goldman_LOA - Gold.pdf").fill(
+    filled = FormWrapper("LOA_Sample.pdf").fill(
         {
-            "Title of Account": name,
-            "Account Number": id,
-            "Print Name and Title": name
+            "company name": cname,
+            "employee name": ename,
+            
         },
     )
-    #output_file_name =  f"documents\\{name}.pdf"
-    output_file_name =  f"{name}.pdf"
+    output_file_name =  f"documents\\{ename}_{cname}.pdf"
     with open(output_file_name, "wb+") as output:
         output.write(filled.read())
 
@@ -1369,7 +1368,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
         sample_button = gr.Button("What are the details of ASN24091600002", elem_classes="gr-buttonbig")
         sample_button1 = gr.Button("What are the active warehouses available", elem_classes="gr-buttonbig")
         sample_button2 = gr.Button("Explain Pre-Receiving Yard Management", elem_classes="gr-buttonbig")
-        sample_button3 = gr.Button("can you generate a histogram chart with item name and customer for warehouse WH1000001", elem_classes="gr-buttonbig")
+        sample_button3 = gr.Button("can you generate a doughnut chart with item name and quantities for warehouse WH1000001", elem_classes="gr-buttonbig")
         sample_button4 = gr.Button("Analyze item name & quantity for different customers in a stacked bar chart for the warehouse WH1000001 & send email to [email protected]", elem_classes="gr-button")
 
     # Chatbot component