Update app.py

app.py

@@ -8,9 +8,69 @@ st.set_page_config(layout="wide")
 API_URL = "https://virtual-eng-expert-api-v1.greensea-b20be511.northeurope.azurecontainerapps.io/api"
 headers = {"Content-Type": "application/json"}
 
-def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
-	return response.json()
+def query(payload, retries=2):
+	"""Query the API with retry logic"""
+	for attempt in range(retries + 1):
+		try:
+			response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
+			response.raise_for_status()  # Raise an exception for bad status codes
+			
+			# Check if response is actually JSON before parsing
+			content_type = response.headers.get('Content-Type', '')
+			if 'application/json' not in content_type:
+				# Response is not JSON - likely HTML error page
+				error_text = response.text[:1000]
+				st.error(f"API returned non-JSON response (Content-Type: {content_type})")
+				st.error(f"Response preview: {error_text[:500]}")
+				with st.expander("Debug: View full response"):
+					st.text(error_text)
+				with st.expander("Debug: View request payload"):
+					st.json(payload)
+				raise ValueError(f"API returned non-JSON response. Status: {response.status_code}")
+			
+			# Try to parse JSON
+			try:
+				return response.json()
+			except ValueError as json_error:
+				# Response claims to be JSON but isn't valid
+				st.error(f"API returned invalid JSON. Response text: {response.text[:500]}")
+				with st.expander("Debug: View request payload"):
+					st.json(payload)
+				raise ValueError(f"Invalid JSON response: {str(json_error)}")
+				
+		except requests.exceptions.HTTPError as e:
+			if attempt < retries:
+				st.warning(f"Attempt {attempt + 1} failed with {e.response.status_code}. Retrying...")
+				continue
+			error_msg = f"API Error {e.response.status_code}"
+			try:
+				# Try to get JSON error if available
+				error_json = e.response.json()
+				if isinstance(error_json, dict):
+					error_msg += f": {error_json.get('error', error_json.get('message', str(error_json)))}"
+				else:
+					error_msg += f": {error_json}"
+			except:
+				# Not JSON, show text response
+				error_text = e.response.text[:500]
+				if error_text:
+					error_msg += f": {error_text}"
+			st.error(error_msg)
+			with st.expander("Debug: View request payload"):
+				st.json(payload)
+			raise
+		except requests.exceptions.Timeout:
+			if attempt < retries:
+				st.warning(f"Attempt {attempt + 1} timed out. Retrying...")
+				continue
+			st.error("Request timed out. The API may be slow or unavailable. Please try again.")
+			raise
+		except requests.exceptions.RequestException as e:
+			if attempt < retries:
+				st.warning(f"Attempt {attempt + 1} failed: {str(e)[:100]}. Retrying...")
+				continue
+			st.error(f"Connection Error: {str(e)}")
+			raise
 
 def generate_response(user,prompt):
     output = query({"namespace":"virtualEngineer","version":1,"messages":[{"role":user, "content": prompt}]})