Update app.py

app.py

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