Update app.py

app.py

@@ -30,7 +30,15 @@ def api_call(prompt, thread_id, model_name, chat_history):
         "chat_history": chat_history  # Include chat history
     }
     response = requests.post(url, headers=headers, json=data)
-    return response.json()["message"]
+    
+    # Log the full response for debugging
+    print("API Response:", response.json())
+    
+    try:
+        return response.json()["message"]
+    except KeyError:
+        print("Key 'message' not found in the response")
+        return "Error: Unable to retrieve the message"
 
 # Gradio interface
 with gr.Blocks() as demo: