Update app.py

app.py

@@ -1,16 +1,15 @@
 import gradio as gr
 import pandas as pd
-from datetime import date
-import pymssql
-
 from datetime import datetime
+import pymssql
 
-start_date = datetime.strptime(start_date, "%Y-%m-%d").date()
-end_date = datetime.strptime(end_date, "%Y-%m-%d").date()
-
-
+# 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()
@@ -25,13 +24,11 @@ def run_procedure(server, database, username, password, start_date, end_date):
             WHERE Document_Procurement_Name = 'EQUITY'
         """
         df = pd.read_sql(query, conn)
-
         conn.close()
 
         if df.empty:
             return "No data found for the given dates.", None
         else:
-            # Save to CSV
             csv_path = "document_procurement_report.csv"
             df.to_csv(csv_path, index=False)
             return df, csv_path
@@ -39,7 +36,7 @@ def run_procedure(server, database, username, password, start_date, end_date):
     except Exception as e:
         return f"Error: {str(e)}", None
 
-# Gradio UI
+# Gradio UI using Textbox for dates (YYYY-MM-DD)
 iface = gr.Interface(
     fn=run_procedure,
     inputs=[
@@ -47,8 +44,8 @@ iface = gr.Interface(
         gr.Textbox(label="Database Name", placeholder="e.g., DocumentAcquisition"),
         gr.Textbox(label="Username"),
         gr.Textbox(label="Password", type="password"),
-        gr.Date(label="Start Date", value=date(2024, 1, 1)),
-        gr.Date(label="End Date", value=date(2024, 1, 7)),
+        gr.Textbox(label="Start Date (YYYY-MM-DD)", value="2024-01-01"),
+        gr.Textbox(label="End Date (YYYY-MM-DD)", value="2024-01-07"),
     ],
     outputs=[
         gr.Dataframe(label="Procedure Output"),