Update app.py

app.py

@@ -4,6 +4,7 @@ from sklearn.ensemble import RandomForestClassifier
 from sklearn.preprocessing import LabelEncoder
 from sklearn.model_selection import train_test_split, GridSearchCV
 import gradio as gr
+import os
 import warnings
 warnings.filterwarnings('ignore')
 
@@ -265,7 +266,11 @@ with gr.Blocks(css="""
     
     def predict_and_show(mood, time_of_day, hunger_level, sentiment, snack_type):
         text, snack = predict_snack(mood, time_of_day, hunger_level, sentiment, snack_type)
-        image_path = f"assets/{snack}.png" if snack in ['suya', 'puff puff'] else None
+        image_path = f"assets/{snack.replace(' ', '_')}.jpeg"
+        if not os.path.exists(image_path):
+            image_path = f"assets/{snack.replace(' ', '_')}.png"  # Check for .png as fallback
+            if not os.path.exists(image_path):
+                image_path = "assets/placeholder.jpeg"  # Final fallback
         return text, image_path
     
     predict_btn.click(