Update app.py

wide mode, 2 column mode.

app.py

@@ -83,6 +83,7 @@ def generate_image(prompt):
         return None
 
 def main():
+    st.set_page_config(layout="wide")
     st.title("Enhanced Waifu Character Generator")
 
     # User inputs
@@ -92,55 +93,55 @@ def main():
         hair_color = st.selectbox("Hair Color", ["Blonde", "Brunette", "Red", "Black", "Blue", "Pink"])
         personality = st.selectbox("Personality", ["Tsundere", "Yandere", "Kuudere", "Dandere", "Genki", "Normal"])
         outfit_style = st.selectbox("Outfit Style", ["School Uniform", "Maid Outfit", "Casual", "Kimono", "Gothic Lolita"])
-    with col2:
         hobbies = st.text_input("Hobbies")
         favorite_food = st.text_input("Favorite Food")
         background_story = st.text_area("Background Story")
         system_prompt = st.text_input("Optional System Prompt", "")
 
-    # Advanced settings
-    with st.expander("Advanced Settings"):
-        temperature = st.slider("Temperature", 0.0, 1.0, 0.9, step=0.05)
-        max_new_tokens = st.slider("Max new tokens", 0, 8192, 512, step=64)
-        top_p = st.slider("Top-p (nucleus sampling)", 0.0, 1.0, 0.95, step=0.05)
-        repetition_penalty = st.slider("Repetition penalty", 1.0, 2.0, 1.0, step=0.05)
+        # Advanced settings
+        with st.expander("Advanced Settings"):
+            temperature = st.slider("Temperature", 0.0, 1.0, 0.9, step=0.05)
+            max_new_tokens = st.slider("Max new tokens", 0, 8192, 512, step=64)
+            top_p = st.slider("Top-p (nucleus sampling)", 0.0, 1.0, 0.95, step=0.05)
+            repetition_penalty = st.slider("Repetition penalty", 1.0, 2.0, 1.0, step=0.05)
 
-    # Initialize session state for generated text and image prompt
-    if "character_description" not in st.session_state:
-        st.session_state.character_description = ""
-    if "image_prompt" not in st.session_state:
-        st.session_state.image_prompt = ""
-    if "image_paths" not in st.session_state:
-        st.session_state.image_paths = []
+        # Initialize session state for generated text and image prompt
+        if "character_description" not in st.session_state:
+            st.session_state.character_description = ""
+        if "image_prompt" not in st.session_state:
+            st.session_state.image_prompt = ""
+        if "image_paths" not in st.session_state:
+            st.session_state.image_paths = []
 
-    # 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)
-            image_prompt = format_prompt_for_image(name, hair_color, personality, outfit_style)
+        # 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)
+                image_prompt = format_prompt_for_image(name, hair_color, personality, outfit_style)
 
-            # 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!")
+                # 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!")
 
-    # 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")
+    with col2:
+        # 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")
 
 if __name__ == "__main__":
     main()