Update app.py

app.py

@@ -308,35 +308,25 @@ def upload_file(files) -> List[str]:
 
 def user_prompt_sanitization(user_prompt:str)->str:
   guide = """
-            You are a Groq advisor. Your task is to collect user prompts and, using the available SQL tables, tailor the questions to the data. Use the table descriptions below to sanitize the user's prompt.
-            
-            **Table Descriptions:**
-            {table_description}
+            You are an AI assistant specializing in database queries. Your task is to interpret user questions about a database and refine them based on the available table structures. Use the following table descriptions to guide your responses:
+
+            Database Schema:
+            {database_schema}
             
-            **Organization Tips for Queries:**
-            - When an ID is requested but not explicitly defined, reference a proper ID from the table.
-              - Example: User prompt: "Get the amount of the purchase with the id 5"
-              - Response: "Get the purchase with the purchase_id 5"
-              - Example: User prompt: "Find the review for product 3"
-              - Response: "Find the review with the product_id 3"
-              - Example: User prompt: "Show details for user 7"
-              - Response: "Show details for the user with the user_id 7"
-            - For numeric or quantifiable attributes:
-              - Example: User prompt: "Get the top 3 purchases"
-              - Response: "Get the 3 purchases with the highest amount"
-              - Example: User prompt: "Show the 5 most recent orders"
-              - Response: "Show the 5 orders with the most recent created_at date"
-              - Example: User prompt: "List the top 10 rated products"
-              - Response: "List the 10 products with the highest rating"
-            - Ensure you do not query a table that does not exist.
+            Instructions:
+            1. Interpret the user's question and identify the relevant tables and columns.
+            2. Refine the query to use correct table and column names as per the schema.
+            3. Ensure all IDs are properly referenced (e.g., user_id instead of just id).
+            4. For quantitative queries (e.g., "top 5"), specify the ordering criteria.
+            5. Infer necessary joins between tables when the query spans multiple tables.
+            6. Provide a clear, concise refinement of the user's query that accurately reflects the database structure.
             
-            **Question:**
-            --------
+            Now, please refine the following user query:
             {user_question}
-            --------
+            
          """
 
-  formatted_guide = guide.format(table_description=table_description,user_question=user_prompt)
+  formatted_guide = guide.format(database_schema=table_description,user_question=user_prompt)
   api_key:str=userdata.get("GROQ_API_KEY")
   client = groq.Groq(api_key=api_key)
   return chat_with_groq(client,formatted_guide,"llama3-70b-8192",None)