Update app.py

app.py

@@ -39,11 +39,12 @@ st.write(f"Valid columns: {valid_columns}")
 
 # Function to generate SQL query using Hugging Face model
 def generate_sql_query(question, table_name, columns):
-    # Simplified prompt to focus on generating valid SQL
+    # Simplified and direct prompt to focus on generating valid SQL
     prompt = f"""
-    You are a SQL expert. Generate a SQL query using the following columns:
+    You are a SQL expert. Generate a SQL query using the columns:
     {columns}.
     Question: {question}
+    Respond only with the SQL query.
     """
     response = llm(prompt, max_new_tokens=50, truncation=True)  # Ensure max tokens are reasonable
     return response[0]['generated_text'].strip()