fix negative feedback

app.py

@@ -214,8 +214,8 @@ async def on_message(message: cl.Message):
                 res = await cl.AskActionMessage(
                     content=f"The query returned {num_rows} rows and {num_cols} columns.\n\nWas this result helpful?",
                     actions=[
-                        cl.Action(name="feedback", payload={"value": "positive"}, label="👍 Good results"),
-                        cl.Action(name="feedback", payload={"value": "negative"}, label="👎 Not what I wanted")
+                        cl.Action(name="feedback_positive", payload={"value": "positive"}, label="👍 Good results"),
+                        cl.Action(name="feedback_negative", payload={"value": "negative"}, label="👎 Not what I wanted")
                     ],
                 ).send()
                 
@@ -275,71 +275,50 @@ async def on_message(message: cl.Message):
         ).send()
 
 # Callback handlers for actions
-@cl.action_callback("feedback")
-async def on_feedback_action(action):
-    """Handle feedback action."""
-    feedback_value = action.payload.get("value")
-    
+@cl.action_callback("feedback_positive")
+async def on_feedback_positive(action):
+    """Handle positive feedback."""
     client = cl.user_session.get("client")
     original_query = cl.user_session.get("original_query")
     generated_sql = cl.user_session.get("generated_sql")
     optimized_sql = cl.user_session.get("optimized_sql")
     
-    if feedback_value == "positive":
-        # Handle positive feedback
-        success = save_feedback_to_bigquery(
-            client, 
-            original_query, 
-            generated_sql, 
-            optimized_sql, 
-            "positive"
-        )
-        
-        if success:
-                await cl.Message(content="Thanks for your positive feedback! I've saved it to improve future responses.", author="SQL Assistant").send()
-        else:
-                await cl.Message(content="Thanks for your feedback! (Note: There was an issue saving it to the database)", author="SQL Assistant").send()
+    # Handle positive feedback
+    success = save_feedback_to_bigquery(
+        client, 
+        original_query, 
+        generated_sql, 
+        optimized_sql, 
+        "positive"
+    )
+    
+    if success:
+        await cl.Message(content="Thanks for your positive feedback! I've saved it to improve future responses.", author="SQL Assistant").send()
+    else:
+        await cl.Message(content="Thanks for your feedback! (Note: There was an issue saving it to the database)", author="SQL Assistant").send()
 
-@cl.action_callback("feedback_bad")
-async def on_feedback_bad(action):
+@cl.action_callback("feedback_negative")
+async def on_feedback_negative(action):
     """Handle negative feedback."""
     # Ask for more detailed feedback
-    res = await cl.AskUserMessage(
-        content="I'm sorry the results weren't what you expected. Could you please provide more details about what was wrong?",
-        author="SQL Assistant",
-        timeout=300,
-        elements=[
-            cl.Textarea(
-                id="feedback_details",
-                label="Your feedback",
-                initial_value="",
-                rows=3
-            )
-        ]
-    ).send()
+    await cl.Message(content="I'm sorry the results weren't what you expected. Please type your feedback about what was wrong.", author="SQL Assistant").send()
     
-    feedback_details = "negative"
-    if res and "feedback_details" in res:
-        feedback_details = f"negative: {res['feedback_details']}"
+    # Set flag to indicate we're awaiting detailed feedback
+    cl.user_session.set("awaiting_feedback", True)
     
     client = cl.user_session.get("client")
     original_query = cl.user_session.get("original_query")
     generated_sql = cl.user_session.get("generated_sql")
     optimized_sql = cl.user_session.get("optimized_sql")
     
-    # Save the feedback to BigQuery
-    success = save_feedback_to_bigquery(
+    # Save initial negative feedback
+    save_feedback_to_bigquery(
         client, 
         original_query, 
         generated_sql, 
         optimized_sql, 
-        feedback_details
+        "negative"
     )
-    
-    if success:
-        await cl.Message(content="Thanks for your detailed feedback! I've saved it to improve future responses.", author="SQL Assistant").send()
-    else:
-        await cl.Message(content="Thanks for your feedback! (Note: There was an issue saving it to the database)", author="SQL Assistant").send()
 
 # This is needed for Chainlit to run properly
 if __name__ == "__main__":