Update app.py

app.py

@@ -15,13 +15,13 @@ def query2(payload):
     response = requests.post(API_URL2, headers=headers2, json=payload)
     return response.json()
 
-def inference_ui(question, context):
+def detect_context_from_question(question):
     output = query({
-        "inputs": f"context for '{question}' is: {context}",
+        "inputs": f"context for '{question}' is:",
     })
-    return output['context']
+    return output
 
-def ask_question_ui(question, context):
+def ask_question(question, context):
     output2 = query2({
         "inputs": {
             "question": question,
@@ -31,14 +31,14 @@ def ask_question_ui(question, context):
     return output2
 
 iface = gr.Interface(
-    fn=ask_question_ui,
+    fn=ask_question,
     inputs=[
-        gr.Textbox("Enter your question", key="question"),
-        gr.Textbox("Enter context", key="context"),
-        gr.Button("Ask", key="ask_button"),
-        gr.Button("AutoContext", key="autocontext_button")
+        gr.Textbox("Enter your question", type="text", key="question"),
+        gr.Textbox("Enter context", type="text", key="context"),
+        "Detect Context", gr.Button(detect_context_from_question, key="detect_context_button"),
+        "Ask", gr.Button("Ask", key="ask_button")
     ],
-    outputs=gr.Textbox("Output"),
+    outputs=gr.Textbox("Output", type="text"),
     live=True
 )