Update app.py

app.py

@@ -6,19 +6,14 @@ import pymssql
 # Function to connect to SQL Server and run stored procedure
 def run_procedure(server, database, username, password, start_date, end_date):
     try:
-        # Convert string dates to datetime.date objects
         start_date = datetime.strptime(start_date, "%Y-%m-%d").date()
         end_date = datetime.strptime(end_date, "%Y-%m-%d").date()
 
-        # Connect to SQL Server using pymssql (no ODBC required)
         conn = pymssql.connect(server=server, user=username, password=password, database=database)
         cursor = conn.cursor()
-
-        # Run the stored procedure with parameters
         cursor.execute("EXEC dbo.Temp_Get_EQUITY_DocumentProcurement %s, %s", (start_date, end_date))
         conn.commit()
 
-        # Now read the output table using pandas
         query = """
             SELECT * FROM [DocumentAcquisition].dbo.Temp_Document_Procurement_Timeline
             WHERE Document_Procurement_Name = 'EQUITY'
@@ -27,16 +22,16 @@ def run_procedure(server, database, username, password, start_date, end_date):
         conn.close()
 
         if df.empty:
-            return "No data found for the given dates.", None
+            return None, None, "No data found for the given dates."
         else:
             csv_path = "document_procurement_report.csv"
             df.to_csv(csv_path, index=False)
-            return df, csv_path
+            return df, csv_path, "Success"
 
     except Exception as e:
-        return f"Error: {str(e)}", None
+        return None, None, f"Error: {str(e)}"
 
-# Gradio UI using Textbox for dates (YYYY-MM-DD)
+# Updated Gradio UI
 iface = gr.Interface(
     fn=run_procedure,
     inputs=[
@@ -50,6 +45,7 @@ iface = gr.Interface(
     outputs=[
         gr.Dataframe(label="Procedure Output"),
         gr.File(label="Download CSV Report"),
+        gr.Textbox(label="Status / Error Message")
     ],
     title="EQUITY Document Procurement Report Generator",
     description="Enter your SQL Server details and date range to run the stored procedure and download the report."