Update app.py

app.py

@@ -85,8 +85,13 @@ def generate_image(prompt):
             st.error("Unexpected result format from the Gradio API.")
             return None
     except Exception as e:
-        st.error(f"Error generating image: {e}")
-        st.write("Full error details:", e)
+        error_message = str(e)
+        match = re.search(r'retry in (\d+:\d+:\d+)', error_message)
+        if match:
+            wait_time = match.group(1)
+            st.error(f"Please wait {wait_time} before you try again. Meanwhile, leave a like if you like the app!")
+        else:
+            st.error(f"Error generating image: {e}")
         return None
 
 def main():