Update app.py

app.py

@@ -137,17 +137,25 @@ def execute_sql(sql_query):
         return str(e)
 
 # 🔹 STEP 4: Final combined response
+
 def response(user_query):
     embeddings, model, student, employee, course = create_metadata_embeddings()
     table_metadata = find_best_fit(embeddings, model, user_query, student, employee, course)
     system_prompt, user_prompt = create_prompt(user_query, table_metadata)
-    sql_query = generate_sql(system_prompt, user_prompt)
+    sql_query = generate_output(system_prompt, user_prompt)
+
+    # Try running the query against the SQLite database
+    try:
+        conn = sqlite3.connect("college.db")  # Make sure college.db is present in your repo
+        cursor = conn.cursor()
+        cursor.execute(sql_query)
+        result = cursor.fetchall()
+        conn.close()
+
+        return f"SQL Query:\n{sql_query}\n\nQuery Result:\n{result}"
+    except Exception as e:
+        return f"SQL Query:\n{sql_query}\n\nQuery Result:\nError: {str(e)}"
 
-    if sql_query:
-        result = execute_sql(sql_query)
-        return f"🧠 SQL Query:\n{sql_query}", result
-    else:
-        return "❌ Couldn't generate a valid SQL query.", None
 
 # 🔹 Gradio UI
 desc = """Ask a natural language question about students, employees, or courses. I'll generate and run a SQL query for you."""