Update app.py

app.py

@@ -15,11 +15,12 @@ def query2(payload):
     response = requests.post(API_URL2, headers=headers2, json=payload)
     return response.json()
 
-def detect_context_from_question(question):
+def detect_context_from_question(question_textbox, context_textbox):
+    question = question_textbox.value
     output = query({
         "inputs": f"context for '{question}' is:",
     })
-    return output
+    context_textbox.value = output['context']
 
 def ask_question(question, context):
     output2 = query2({
@@ -33,12 +34,14 @@ def ask_question(question, context):
 iface = gr.Interface(
     fn=ask_question,
     inputs=[
-        gr.Textbox("Enter your question", type="text"),
-        gr.Textbox("Enter context", type="text"),
-        gr.Button(detect_context_from_question)
+        gr.Textbox("Enter your question", type="text", name="a"),
+        gr.Textbox("Enter context", type="text", name="b"),
+        gr.Button(detect_context_from_question, text="Detect Context", name="detect_button"),
+        gr.Button("Ask", name="ask_button")
     ],
-    outputs=gr.Textbox("Output", type="text"),
+    outputs=gr.Textbox("Output", type="text", name="b"),  # Show context in the same Textbox as input
     live=True
 )
 
 iface.launch()
+