Update app.py

app.py

@@ -3,9 +3,6 @@ from huggingface_hub import InferenceClient
 from gradio_client import Client
 import re
 
-# Set page configuration
-st.set_page_config(layout="wide")
-
 # Load custom CSS
 with open('style.css') as f:
     st.markdown(f'<style>{f.read()}</style>', unsafe_allow_html=True)
@@ -90,6 +87,7 @@ def generate_image(prompt):
         return None
 
 def main():
+    st.set_page_config(layout="wide")
     st.title("Free Waifu Character Generator")
     st.write("""
     Welcome to the Free Waifu Character Generator! This app allows you to create a unique waifu with just one click. 
@@ -142,46 +140,41 @@ def main():
 
         # Generate button
         if st.button("Generate Waifu"):
-            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 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 image
+        # 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)
         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()