Update app.py

app.py

@@ -39,8 +39,9 @@ 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
     prompt = f"""
-    Generate a valid SQL query using the following columns:
+    You are a SQL expert. Generate a SQL query using the following columns:
     {columns}.
     Question: {question}
     """
@@ -138,7 +139,8 @@ def process_input():
                             assistant_response = f"Error executing SQL query: {e}"
                             st.session_state.history.append({"role": "assistant", "content": assistant_response})
                     else:
-                        st.session_state.history.append({"role": "assistant", "content": "Generated text is not a valid SQL query."})
+                        # If generated text is not valid SQL, provide feedback to the user
+                        st.session_state.history.append({"role": "assistant", "content": "Generated text is not a valid SQL query. Please try rephrasing your question."})
             else:  # INSIGHTS category
                 # Generate dataset summary
                 dataset_summary = generate_dataset_summary(data)