Update app.py

app.py

@@ -85,13 +85,8 @@ def generate_image(prompt):
             st.error("Unexpected result format from the Gradio API.")
             return None
     except Exception as 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}")
+        st.error(f"Error generating image: {e}")
+        st.write("Full error details:", e)
         return None
 
 def main():
@@ -160,8 +155,7 @@ def main():
                     )
 
                     # Generate character description
-                    character_description = generate_text(description_prompt, temperature, max_new_tokens, top_p, repetition_penalty)
-                    st.session_state.character_description = character_description
+                    st.session_state.character_description = generate_text(description_prompt, temperature, max_new_tokens, top_p, repetition_penalty)
                     
                     # Generate image prompt
                     st.session_state.image_prompt = generate_text(image_prompt, temperature, max_new_tokens, top_p, repetition_penalty)