Update app.py

app.py

@@ -1,15 +1,15 @@
 import os
 import gradio as gr
-from sqlalchemy import text, inspect
+from sqlalchemy import text, inspect, create_engine
 from smolagents import tool, CodeAgent, HfApiModel
 import pandas as pd
 import tempfile
 from database import engine, initialize_database
 
-# Ensure the database initializes
+# Ensure the database initializes (won't crash if empty)
 initialize_database()
 
-# Function to execute SQL script from uploaded file
+# Function to execute an uploaded SQL script
 def execute_sql_script(file_path):
     """
     Executes an uploaded SQL file to initialize the database.
@@ -25,14 +25,14 @@ def execute_sql_script(file_path):
             with open(file_path, "r") as f:
                 sql_script = f.read()
                 con.execute(text(sql_script))
-        return "SQL file executed successfully."
+        return "SQL file executed successfully. Database updated."
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Function to fetch table names dynamically
+# Function to get table names dynamically
 def get_table_names():
     """
-    Returns a list of all tables in the database.
+    Returns a list of tables available in the database.
 
     Returns:
         list: List of table names.
@@ -40,7 +40,7 @@ def get_table_names():
     inspector = inspect(engine)
     return inspector.get_table_names()
 
-# Function to fetch table schema dynamically
+# Function to get table schema dynamically
 def get_table_schema(table_name):
     """
     Returns a list of column names for a given table.
@@ -55,10 +55,10 @@ def get_table_schema(table_name):
     columns = inspector.get_columns(table_name)
     return [col["name"] for col in columns]
 
-# Function to fetch table data dynamically
+# Function to fetch data dynamically from any table
 def get_table_data(table_name):
     """
-    Retrieves all rows from the specified table as a Pandas DataFrame.
+    Retrieves all rows from a specified table as a Pandas DataFrame.
 
     Args:
         table_name (str): Name of the table.
@@ -80,17 +80,17 @@ def get_table_data(table_name):
     except Exception as e:
         return pd.DataFrame({"Error": [str(e)]})
 
-# SQL Execution Tool (FIXED Docstring for `smolagents`)
+# SQL Execution Tool for natural language queries
 @tool
 def sql_engine(query: str) -> str:
     """
     Executes an SQL SELECT query and returns the results.
 
-    Args:
-        query (str): The SQL query string to execute.
+    Parameters:
+        query (str): The SQL query string to execute. Only SELECT queries are allowed.
 
     Returns:
-        str: The formatted query results, or an error message if the query fails.
+        str: A formatted string containing the query results, or an error message if the query fails.
     """
     try:
         with engine.connect() as con:
@@ -101,16 +101,16 @@ def sql_engine(query: str) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Function to generate and execute SQL queries dynamically
+# Function to handle natural language to SQL conversion
 def query_sql(user_query: str) -> str:
     """
-    Processes a user’s natural language query and generates an SQL query dynamically.
+    Converts a user's natural language query into an SQL query.
 
     Args:
         user_query (str): The question asked by the user.
 
     Returns:
-        str: SQL query results or an error message.
+        str: The results of the executed SQL query.
     """
     tables = get_table_names()
     if not tables:
@@ -144,10 +144,10 @@ def handle_query(user_input: str) -> str:
     """
     return query_sql(user_input)
 
-# Function to handle SQL file uploads
+# Function to handle SQL file uploads and execute them
 def handle_file_upload(file):
     """
-    Handles file upload, executes SQL, and updates database schema dynamically.
+    Handles SQL file upload, executes SQL, and updates database schema.
 
     Args:
         file (File): Uploaded SQL file.
@@ -197,3 +197,4 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
 
+