Spaces:

cha0smagick
/

Mago_la_ascencion_character_generator

Running

App Files Files Community

cha0smagick commited on Sep 24, 2024

Commit

387c702

verified ·

1 Parent(s): f8bf09d

Update app.py

Browse files

Files changed (1) hide show

app.py +38 -111

app.py CHANGED Viewed

@@ -131,7 +131,6 @@ def generate_description():
         "They can charm their way out of most situations with their charisma.",
         "Often reserved, they observe more than they speak, learning the secrets around them."
     ]
     return random.choice(character_traits)
 # Función para generar un background de personaje basado en sus atributos
@@ -146,6 +145,28 @@ def generate_background():
     ]
     return random.choice(backgrounds)
 # Inicializar la aplicación
 st.title("Character Creator for MAGE")
@@ -161,116 +182,22 @@ if st.session_state.character_sheet:
     st.subheader("Character Sheet")
     st.write(character_sheet)
-    def download_html(character_sheet):
-        html = f"""
-        <html>
-        <head>
-            <style>
-                body {{
-                    font-family: Arial, sans-serif;
-                }}
-                table {{
-                    border-collapse: collapse;
-                    width: 100%;
-                }}
-                th, td {{
-                    border: 1px solid #000;
-                    padding: 8px;
-                    text-align: left;
-                }}
-                th {{
-                    background-color: #f2f2f2;
-                }}
-                h1, h2, h3 {{
-                    color: #333;
-                }}
-            </style>
-        </head>
-        <body>
-            <h1>MAGE Character Sheet</h1>
-            <h2>Basic Information</h2>
-            <p>Name: {character_sheet["Name"]}</p>
-            <p>Nature: {character_sheet["Nature"]}</p>
-            <p>Demeanor: {character_sheet["Demeanor"]}</p>
-            <p>Essence: {character_sheet["Essence"]}</p>
-            <p>Affiliation: {character_sheet["Affiliation"]}</p>
-            <p>Sect: {character_sheet["Sect"]}</p>
-            <p>Concept: {character_sheet["Concept"]}</p>
-            <p>Background: {character_sheet["Background"]}</p>
-            <p>Description: {character_sheet["Description"]}</p>
-            <h2>Attributes</h2>
-            <table>
-                <tr><th>Attribute</th><th>Value</th></tr>
-                {''.join([f"<tr><td>{attr}</td><td>{value}</td></tr>" for attr, value in character_sheet["Attributes"].items()])}
-            </table>
-            <h2>Talents</h2>
-            <table>
-                <tr><th>Talent</th><th>Value</th></tr>
-                {''.join([f"<tr><td>{talent}</td><td>{value}</td></tr>" for talent, value in character_sheet["Talents"].items()])}
-            </table>
-            <h2>Skills</h2>
-            <table>
-                <tr><th>Skill</th><th>Value</th></tr>
-                {''.join([f"<tr><td>{skill}</td><td>{value}</td></tr>" for skill, value in character_sheet["Skills"].items()])}
-            </table>
-            <h2>Knowledges</h2>
-            <table>
-                <tr><th>Knowledge</th><th>Value</th></tr>
-                {''.join([f"<tr><td>{knowledge}</td><td>{value}</td></tr>" for knowledge, value in character_sheet["Knowledges"].items()])}
-            </table>
-            <h2>Spheres</h2>
-            <table>
-                <tr><th>Sphere</th><th>Value</th></tr>
-                {''.join([f"<tr><td>{sphere}</td><td>{value}</td></tr>" for sphere, value in character_sheet["Spheres"].items()])}
-            </table>
-            <h2>Advantages</h2>
-            <table>
-                <tr><th>Advantage</th><th>Value</th></tr>
-                {''.join([f"<tr><td>{advantage}</td><td>{value}</td></tr>" for advantage, value in character_sheet["Advantages"]["Backgrounds"].items()])}
-            </table>
-            <h2>Health</h2>
-            <table>
-                <tr><th>Health Level</th><th>Value</th></tr>
-                {''.join([f"<tr><td>{health}</td><td>{value}</td></tr>" for health, value in character_sheet["Health"].items()])}
-            </table>
-            <h2>Backgrounds</h2>
-            <table>
-                <tr><th>Background</th><th>Value</th></tr>
-                {''.join([f"<tr><td>{background}</td><td>{value}</td></tr>" for background, value in character_sheet["Backgrounds"].items()])}
-            </table>
-            <h2>Other Traits</h2>
-            <table>
-                <tr><th>Quirk</th><th>Value</th></tr>
-                {''.join([f"<tr><td>{trait}</td><td>{value}</td></tr>" for trait, value in character_sheet["Other Traits"].items()])}
-            </table>
-        </body>
-        </html>
-        """
-        return html
-    st.markdown("### Download Character Sheet as HTML")
-    html_content = download_html(character_sheet)
-    b64 = base64.b64encode(html_content.encode()).decode()
-    href = f'<a href="data:text/html;base64,{b64}" download="character_sheet.html">Download Character Sheet</a>'
-    st.markdown(href, unsafe_allow_html=True)
-# Define la función principal para generar imágenes utilizando freeGPT
-async def main():
-    prompt = f"""
-    Create a detailed image of the following character based on these attributes:
-    Name: {character_sheet['Name']},
-    Nature: {character_sheet['Nature']},
-    Demeanor: {character_sheet['Demeanor']},
-    Essence: {character_sheet['Essence']},
-    Affiliation: {character_sheet['Affiliation']},
-    Sect: {character_sheet['Sect']},
-    Concept: {character_sheet['Concept']}.
-    Description: {character_sheet['Description']}.
-    """
-    await freeGPT.generate_image(prompt)
-# Llamar a la función principal si se ha generado un personaje
-if st.session_state.character_sheet:
-    st.subheader("Generate Image")
-    if st.button("Generate Image Based on Character"):
-        await main()

         "They can charm their way out of most situations with their charisma.",
         "Often reserved, they observe more than they speak, learning the secrets around them."
     ]
     return random.choice(character_traits)
 # Función para generar un background de personaje basado en sus atributos
     ]
     return random.choice(backgrounds)
+# Función para generar la imagen del personaje usando freeGPT
+async def generate_image(prompt):
+    try:
+        response = await getattr(freeGPT, "prodia").Generation().create(prompt)
+        image = Image.open(BytesIO(response))
+        image.show()
+        return image
+    except Exception as e:
+        st.error(f"Error generating image: {e}")
+        return None
+# Función para generar el prompt de la imagen basado en la información del personaje
+def create_image_prompt(character):
+    prompt = (
+        f"Create an image of a character named {character['Name']}, who is a {character['Nature']} "
+        f"with a demeanor that is {character['Demeanor']}. Their essence is {character['Essence']} "
+        f"and they are affiliated with a {character['Affiliation']} in the {character['Sect']} sect. "
+        f"Conceptually, they are {character['Concept']}. "
+        f"Background: {character['Background']}. Description: {character['Description']}."
+    )
+    return prompt
 # Inicializar la aplicación
 st.title("Character Creator for MAGE")
     st.subheader("Character Sheet")
     st.write(character_sheet)
+    # Generar y mostrar imagen del personaje
+    prompt = create_image_prompt(character_sheet)
+    st.write(f"Prompt for image: {prompt}")
+    if st.button("Generate Character Image"):
+        image = st.session_state.character_image = await generate_image(prompt)
+        if image:
+            st.image(image, caption="Character Image")
+    def download_html(character_sheet):
+        # (función para descargar la hoja del personaje en HTML)
+        # Código original de la función download_html aquí...
+    # Botón para descargar el HTML
+    if st.button("Download Character Sheet as HTML"):
+        html = download_html(character_sheet)
+        b64 = base64.b64encode(html.encode()).decode()
+        href = f'<a href="data:text/html;base64,{b64}" download="character_sheet.html">Download your character sheet!</a>'
+        st.markdown(href, unsafe_allow_html=True)