Update app.py

app.py

@@ -192,29 +192,35 @@ def submit_feedback(question, model_answer, feedback, correct_answer):
 # === Gradio UI ===
 with gr.Blocks() as demo:
     gr.Markdown("## 🧮 Math Question Answering with DSPy + Feedback")
-    
-    with gr.Tab("Ask a Math Question"):
+
+    with gr.Tab("Ask a Math Question & Submit Feedback"):
         with gr.Row():
             question_input = gr.Textbox(label="Enter your math question", lines=2)
+            submit_btn = gr.Button("Get Answer")
+
         gr.Markdown("### 🧠 Answer:")
         answer_output = gr.Markdown()
 
-        #answer_output = gr.Markdown(label="Answer")
+        # Hidden fields to hold question and answer for feedback inputs
         hidden_q = gr.Textbox(visible=False)
         hidden_a = gr.Textbox(visible=False)
-        submit_btn = gr.Button("Get Answer")
+
+        # Connect submit button to ask_question function
         submit_btn.click(fn=ask_question, inputs=[question_input], outputs=[answer_output, hidden_q, hidden_a])
 
-    with gr.Tab("Submit Feedback"):
-        gr.Markdown("### Was the answer helpful?")
-        fb_question = gr.Textbox(label="Original Question")
-        fb_answer = gr.Textbox(label="Model's Answer")
-        fb_like = gr.Radio(["👍", "👎"], label="Your Feedback")
-        fb_correct = gr.Textbox(label="Correct Answer (optional)")
+        gr.Markdown("### 📝 Submit Feedback")
+        fb_like = gr.Radio(["👍", "👎"], label="Was the answer helpful?")
+        fb_correct = gr.Textbox(label="Correct Answer (optional) or Comments")
         fb_submit_btn = gr.Button("Submit Feedback")
         fb_status = gr.Textbox(label="Status", interactive=False)
-        fb_submit_btn.click(fn=submit_feedback,
-                            inputs=[fb_question, fb_answer, fb_like, fb_correct],
-                            outputs=[fb_status])
+
+        # Feedback submit button uses hidden fields + feedback inputs
+        fb_submit_btn.click(
+            fn=submit_feedback,
+            inputs=[hidden_q, hidden_a, fb_like, fb_correct],
+            outputs=[fb_status]
+        )
+
+
 
 demo.launch(share=True, debug=True)