Update app.py

app.py

@@ -33,12 +33,17 @@ input_data = {
 
 # Make prediction when the "Predict" button is clicked
 if st.button("Predict"):
-    response = requests.post(
+    try:
+        response = requests.post(
         "https://maddykan101-SalesPredictionBackend.hf.space/v1/prediction",
         json=input_data
-    )
-    if response.status_code == 200:
-        prediction = response.json()['Predicted Sales ']  # Adjust key if needed
-        st.success(f"Predicted Sales: {prediction}")
-    else:
-        st.error(f"Prediction failed: {response.status_code}")
+        )
+        if response.status_code == 200:
+            prediction = response.json()['Predicted Sales ']  # Adjust key if needed
+            st.success(f"Predicted Sales: {prediction}")
+        else:
+            st.error(f"Prediction failed: {response.status_code}")
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+        
+