Update app.py

app.py

@@ -15,9 +15,9 @@ def query2(payload):
     response = requests.post(API_URL2, headers=headers2, json=payload)
     return response.json()
 
-def inference_ui(question):
+def inference_ui(question, context):
     output = query({
-        "inputs": f"context for '{question}' is:",
+        "inputs": f"context for '{question}' is: {context}",
     })
     return output['context']
 
@@ -30,9 +30,16 @@ def ask_question_ui(question, context):
     })
     return output2
 
-iface_context = gr.Interface(fn=inference_ui, inputs="text", outputs="text", live=True)
-iface_ask_question = gr.Interface(fn=ask_question_ui, inputs=["text", "text"], outputs="text", live=True)
-
-iface_context.launch(share=True)
-iface_ask_question.launch(share=True)
-
+iface = gr.Interface(
+    fn=ask_question_ui,
+    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")
+    ],
+    outputs=gr.Textbox("Output"),
+    live=True
+)
+
+iface.launch()