Update app.py

app.py

@@ -140,41 +140,46 @@ def main():
 
         # Generate button
         if st.button("Generate Waifu"):
-            with st.spinner("Generating waifu character..."):
-                description_prompt = format_prompt_for_description(
-                    name, hair_color, personality, outfit_style, hobbies, favorite_food, background_story,
-                    eye_color=eye_color, height=height, age=age, favorite_quote=favorite_quote, strengths=strengths,
-                    weaknesses=weaknesses, goals=goals
-                )
-                image_prompt = format_prompt_for_image(
-                    name, hair_color, personality, outfit_style,
-                    background_setting=background_setting, pose=pose
-                )
-
-                # Generate 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)
-                
-                # Generate image from image prompt
-                st.session_state.image_paths = generate_image(st.session_state.image_prompt)
-                
-                st.success("Waifu character generated!")
+            with right_col:
+                with st.spinner("Generating waifu character..."):
+                    description_prompt = format_prompt_for_description(
+                        name, hair_color, personality, outfit_style, hobbies, favorite_food, background_story,
+                        eye_color=eye_color, height=height, age=age, favorite_quote=favorite_quote, strengths=strengths,
+                        weaknesses=weaknesses, goals=goals
+                    )
+                    image_prompt = format_prompt_for_image(
+                        name, hair_color, personality, outfit_style,
+                        background_setting=background_setting, pose=pose
+                    )
+
+                    # Generate 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)
+                    
+                    # Generate image from image prompt
+                    st.session_state.image_paths = generate_image(st.session_state.image_prompt)
+                    
+                    st.success("Waifu character generated!")
 
     with right_col:
         st.header("Generated Waifu")
-        # Display the generated character and image prompt
-        if st.session_state.character_description:
-            st.subheader("Generated Waifu Character")
-            st.write(st.session_state.character_description)
-        if st.session_state.image_prompt:
-            st.subheader("Image Prompt")
-            st.write(st.session_state.image_prompt)
+        # Display the generated image
         if st.session_state.image_paths:
             st.subheader("Generated Image")
             for image_path in st.session_state.image_paths:
                 st.image(image_path, caption="Generated Waifu Image")
 
+        # Display the generated image prompt
+        if st.session_state.image_prompt:
+            st.subheader("Image Prompt")
+            st.write(st.session_state.image_prompt)
+
+        # Display the generated character description
+        if st.session_state.character_description:
+            st.subheader("Generated Waifu Character")
+            st.write(st.session_state.character_description)
+
 if __name__ == "__main__":
     main()