Update app.py

app.py

@@ -92,7 +92,11 @@ def get_bot_response(adapter_id, prompt, state, bot_index):
         "Authorization": f"Bearer {os.environ.get('PREDIBASE_TOKEN')}"
     }
     response = requests.post("https://serving.app.predibase.com/79957f/deployments/v2/llms/mistral-7b/generate", json=payload, headers=headers)
-    return response.json()['generated_text'].split('### Instruction')[0]
+    try:
+        response_text = response.json()['generated_text']
+    except (KeyError, requests.exceptions.JSONDecodeError):
+        response_text = "Sorry, I couldn't generate a response."
+    return response_text.split('### Instruction')[0]
 def chat_with_bots(user_input, state):
     # Use existing bot order from state if available, otherwise shuffle and initialize
     if 'last_bots' not in state or not state['last_bots']: