add handle_query

app.py

@@ -11,7 +11,7 @@ from database import engine, receipts
 def sql_engine(query: str) -> str:
     """
     Executes an SQL query on the 'receipts' table and returns formatted results.
-    
+
     Args:
         query: The SQL query to execute.
 
@@ -25,7 +25,7 @@ def sql_engine(query: str) -> str:
         if not rows:
             return "No results found."
 
-        # Convert results into a readable string format
+        # Convert query results into a clean, readable format
         return "\n".join([", ".join(map(str, row)) for row in rows])
 
     except Exception as e:
@@ -70,26 +70,19 @@ def query_sql(user_query: str) -> str:
     # Log the SQL query result
     print(f"SQL Query Result: {result}")
 
-    return result  # Return the final result, NOT the generated SQL
-
-
-    # Generate SQL query using the provided schema
-    generated_sql = agent.run(f"{schema_info} Convert this request into SQL: {user_query}")
-
-    # Log the generated SQL for debugging
-    print(f"Generated SQL: {generated_sql}")
-
-    # Ensure we only execute valid SELECT queries
-    if not generated_sql.strip().lower().startswith(("select", "show", "pragma")):
-        return "Error: Only SELECT queries are allowed."
+    return result  # Return only the final query result, NOT the generated SQL
 
-    # Execute the SQL query and return the result
-    result = sql_engine(generated_sql)
+def handle_query(user_input: str) -> str:
+    """
+    Calls query_sql, captures the output, and directly returns it to the UI.
 
-    # Log the SQL query result
-    print(f"SQL Query Result: {result}")
+    Args:
+        user_input: The user's natural language question.
 
-    return result 
+    Returns:
+        The SQL query result as a plain string to be displayed in the UI.
+    """
+    return query_sql(user_input)  # Directly return the processed result
 
 # Initialize CodeAgent to generate SQL queries from natural language
 agent = CodeAgent(
@@ -97,9 +90,9 @@ agent = CodeAgent(
     model=HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct"),
 )
 
-# Define Gradio interface
+# Define Gradio interface using handle_query instead of query_sql
 demo = gr.Interface(
-    fn=query_sql,
+    fn=handle_query,  # Call handle_query to return the final SQL output
     inputs=gr.Textbox(label="Enter your query in plain English"),
     outputs=gr.Textbox(label="Query Result"),
     title="Natural Language to SQL Executor",