Update app.py

app.py

@@ -15,19 +15,24 @@ def query2(payload):
     response = requests.post(API_URL2, headers=headers2, json=payload)
     return response.json()
 
-def get_context(question):
-    payload = {"inputs": question}
+def get_context_func(question, context_output):
+    payload = {"question": question}
     result = query(payload)
-    return result['answer']
+    context_output.update(result['answer'])
 
-def ask_ai(question, context):
+def ask_ai(question, context, answer_output):
     payload = {"question": question, "context": context}
     result = query2(payload)
-    return result['answer']
+    answer_output.update(result['answer'])
 
 iface = gr.Interface(
     fn=ask_ai,
-    inputs=[gr.Textbox("Question"), gr.Button("get_context", get_context)],
+    inputs=[
+        gr.Textbox("Question"),
+        gr.Textbox("Context"),
+        gr.Button("get_context", get_context_func),
+        gr.Button("ask_ai")
+    ],
     outputs=gr.Textbox("Answer"),
 )