Update app.py

app.py

@@ -95,16 +95,16 @@ def query_model(question, history):
     """
     Process a question, find relevant information, and generate a response.
     """
-    global question_g
-    question_g = question
+    # global question_g
+    # question_g = question
     if question == "":
         return "What the sigma! You didn't ask a question. Feel free to ask me anything about the topics listed above."
     relevant_segment = find_relevant_segment(question, segments)
     if not relevant_segment:
         return "Unfortunately I cannot answer your question..😔 Try refining your question so I can try again."
-    global response_g
-    response_g = generate_response(question, relevant_segment)
-    return response_g
+    # global response_g
+    response = generate_response(question, relevant_segment)
+    return response
 
 # Define the welcome message and specific topics the chatbot can provide information about
 welcome_message = """
@@ -143,8 +143,8 @@ with gr.Blocks(theme='gradio/soft') as demo:
     with gr.Row():
         flag_btn = gr.Button("Flag")
 
-    callback.setup([response_g, question_g], "flagged_data_points")
-    flag_btn.click(lambda *args: callback.flag(args), [response_g, question_g], None, preprocess=False)
+    callback.setup([chatbot], "flagged_data_points")
+    flag_btn.click(lambda *args: callback.flag(args), [chatbot], None, preprocess=False)
 
 # Launch the Gradio app to allow user interaction
 demo.launch(share=True)