Update app.py

app.py

@@ -81,9 +81,14 @@ if password == st.secrets["real_password"]:
             res = get_response(vision_message)
 
         res_text = ""
-        for chunk in res:
-            res_text += chunk.text
+        try:
+            for chunk in res:
+                res_text += chunk.text
+                res_area.markdown(res_text)
+        except:
+            res_text += res.prompt_feedback
             res_area.markdown(res_text)
+            
 
         messages.append({"role": "model", "parts": [res_text]})
 else: