Update app.py

app.py

@@ -3,6 +3,7 @@ import pickle
 from pydantic import BaseModel
 import streamlit as st
 import requests
+from requests.exceptions import ConnectionError
 
 app = FastAPI()
 
@@ -62,14 +63,23 @@ if __name__ == "__main__":
 
     submit = st.button("Predict")
     if submit:
-        payload = {"Sepal_Length": sepal_length, "Sepal_Width": sepal_width, "Petal_Length": petal_length, "Petal_Width": petal_width}
+        payload = {
+            "Sepal_Length": sepal_length,
+            "Sepal_Width": sepal_width,
+            "Petal_Length": petal_length,
+            "Petal_Width": petal_width
+        }
         prediction = st.empty()
         with st.spinner("Predicting..."):
-            response = requests.post("http://localhost:8000/predict", json=payload)
-            if response.status_code == 200:
-                prediction_result = response.json()
-                prediction.success(f"Prediction: {prediction_result['prediction']}")
-            else:
-                prediction.error("Prediction failed.")
+            try:
+                response = requests.post("http://localhost:8000/predict", json=payload)
+                if response.status_code == 200:
+                    prediction_result = response.json()
+                    prediction.success(f"Prediction: {prediction_result['prediction']}")
+                else:
+                    prediction.error("Prediction failed.")
+            except ConnectionError as e:
+                prediction.error("Failed to connect to the server. Make sure the FastAPI server is running.")
+                print(str(e))
 
     uvicorn_proc.kill()