Update app.py

app.py

@@ -39,17 +39,24 @@ if submitted:
         "Store_Type": store_type
     }]
 
-    # Send request to your Flask API
     try:
         response = requests.post(
             "https://Mahendra87-IP-SuperMart.hf.space/predict",
-            json=input_payload
+            json=input_payload,
+            timeout=10
         )
+
         if response.status_code == 200:
             result = response.json()
-            prediction = result["prediction"][0]
-            st.success(f"Predicted Product Sales: ₹{round(prediction, 2)}")
+            st.write("API response:", result)  # Helpful for debugging
+
+            if "prediction" in result:
+                prediction = result["prediction"][0]
+                st.success(f"Predicted Product Sales: ₹{round(prediction, 2)}")
+            else:
+                st.error("'prediction' key not found in API response.")
         else:
-            st.error("Prediction failed. Please check API or input.")
+            st.error(f"Prediction failed. Status code: {response.status_code}")
+            st.text(response.text)
     except Exception as e:
         st.error(f"API request failed: {e}")