Update app.py

This updated code includes additional character attributes, personality details, and more detailed image generation options, giving users a richer and more customizable experience.

app.py

@@ -11,13 +11,16 @@ with open('style.css') as f:
 text_client = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.1")
 image_client = Client("Boboiazumi/animagine-xl-3.1")
 
-def format_prompt_for_description(name, hair_color, personality, outfit_style, hobbies, favorite_food, background_story):
-    prompt = f"Create a waifu character named {name} with {hair_color} hair, a {personality} personality, and wearing a {outfit_style}. "
-    prompt += f"Her hobbies include {hobbies}. Her favorite food is {favorite_food}. Here is her background story: {background_story}."
+def format_prompt_for_description(name, hair_color, eye_color, height, age, personality, outfit_style, hobbies, favorite_food, favorite_quote, strengths, weaknesses, goals, background_story):
+    prompt = f"Create a waifu character named {name} with {hair_color} hair, {eye_color} eyes, a {personality} personality, and wearing a {outfit_style}. "
+    prompt += f"She is {height} tall and {age} years old. Her hobbies include {hobbies}. Her favorite food is {favorite_food}. "
+    prompt += f"Her favorite quote is: '{favorite_quote}'. She has the following strengths: {strengths}. Her weaknesses include {weaknesses}. "
+    prompt += f"Her goals are: {goals}. Here is her background story: {background_story}."
     return prompt
 
-def format_prompt_for_image(name, hair_color, personality, outfit_style):
-    prompt = f"Generate an image prompt for a waifu character named {name} with {hair_color} hair, a {personality} personality, and wearing a {outfit_style}."
+def format_prompt_for_image(name, hair_color, eye_color, personality, outfit_style, background_setting, pose):
+    prompt = f"Generate an image prompt for a waifu character named {name} with {hair_color} hair, {eye_color} eyes, a {personality} personality, and wearing a {outfit_style}. "
+    prompt += f"She is in a {background_setting} setting, posing in a {pose}."
     return prompt
 
 def clean_generated_text(text):
@@ -83,19 +86,33 @@ def generate_image(prompt):
         return None
 
 def main():
-    st.title("Enhanced Waifu Character Generator")
+    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. 
+    Generate a detailed bio/background story, a prompt for an image, and the image itself. 
+    It's the best free waifu generator out there, providing everything you need to bring your waifu to life.
+    """)
 
     # User inputs
     col1, col2 = st.columns(2)
     with col1:
         name = st.text_input("Name of the Waifu")
         hair_color = st.selectbox("Hair Color", ["Blonde", "Brunette", "Red", "Black", "Blue", "Pink"])
+        eye_color = st.selectbox("Eye Color", ["Blue", "Green", "Brown", "Hazel", "Red", "Purple"])
+        height = st.text_input("Height (e.g., 5'6\")")
+        age = st.text_input("Age")
         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")
+        favorite_quote = st.text_input("Favorite Quote")
+        strengths = st.text_input("Strengths")
+        weaknesses = st.text_input("Weaknesses")
+        goals = st.text_input("Goals")
         background_story = st.text_area("Background Story")
+        background_setting = st.selectbox("Background Setting", ["School", "City", "Fantasy World", "Beach", "Park"])
+        pose = st.selectbox("Pose", ["Standing", "Sitting", "Action Pose", "Lying Down"])
         system_prompt = st.text_input("Optional System Prompt", "")
 
     # Advanced settings
@@ -116,8 +133,11 @@ 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)
-            image_prompt = format_prompt_for_image(name, hair_color, personality, outfit_style)
+            description_prompt = format_prompt_for_description(
+                name, hair_color, eye_color, height, age, personality, outfit_style, 
+                hobbies, favorite_food, favorite_quote, strengths, weaknesses, goals, background_story
+            )
+            image_prompt = format_prompt_for_image(name, hair_color, eye_color, personality, outfit_style, background_setting, pose)
 
             # Generate character description
             st.session_state.character_description = generate_text(description_prompt, temperature, max_new_tokens, top_p, repetition_penalty)