Update app.py

app.py

@@ -31,13 +31,13 @@ with col_right:
 
 # Create JSON payload
 payload = {
-    #"Product_Id": product_id,
+    #"Product_ID": product_id,
     "Product_Weight": weight,
     "Product_Sugar_Content": sugar_level,
     "Product_Allocated_Area": allocated_area,
     "Product_Type": category,
     "Product_MRP": mrp,
-    #"Store_Id": store_id,
+    #"Store_ID": store_id,
     "Store_Establishment_Year": establishment_year,
     "Store_Size": store_size,
     "Store_Location_City_Type": city_tier,
@@ -50,13 +50,19 @@ if st.button("Get Sales Prediction"):
         try:
             api_url = "https://mohith96-SuperKartBackend.hf.space/predict"
             response = requests.post(api_url, json=payload, timeout=10)
+
             if response.status_code == 200:
                 result = response.json()
-                if "Predicted_Sales_Product" in result:
-                    st.success(f"Predicted Sales: $ {result['Predicted_Sales_Product']:.2f}")
+                predicted_value = result.get("Predicted_Sales") or result.get("Predicted_Sales_Product")
+
+                if predicted_value is not None:
+                    st.success(f"Predicted Sales: $ {predicted_value:.2f}")
                 else:
-                    st.warning(f"Unexpected response: {result}")
+                    st.warning(f"Unexpected response format: {result}")
             else:
                 st.error(f"API returned {response.status_code}: {response.text}")
+
+        except requests.exceptions.RequestException as req_err:
+            st.error(f"API request failed: {req_err}")
         except Exception as error:
-            st.error(f"Failed to reach API: {error}")
+            st.error(f"Unexpected error: {error}")