Update app.py

app.py

@@ -24,7 +24,7 @@ ambientazioni = {
     "Sacro Romano Impero": {
         "nome": "Sacro Romano Impero",
         "stile_immagine": (
-            "Highly detailed, painterly style with a historical yet stylized aesthetic. "
+            "In the italian Holy Roman Empire, Highly detailed, painterly style with a historical yet stylized aesthetic. "
             "Rich textures, ornate patterns, and a color palette dominated by imperial gold, "
             "deep red, and aged marble tones. Inspired by medieval European art, "
             "gothic illuminations, and the grandeur of cathedrals. "
@@ -92,39 +92,46 @@ ambientazioni = {
     "Cowboy del Far West": {
         "nome": "Cowboy del Far West",
         "stile_immagine": (
-            "Uno stile che richiama i paesaggi desertici, le praterie sconfinate e i tramonti infuocati. "
-            "Elementi iconici come cappelli da cowboy, revolver e diligenze, uniti a colori terrosi e caldi. "
-            "Una composizione che unisce il romanticismo dell'Ovest selvaggio a un design chiaro e narrativo, "
-            "ideale per illustrare scene di duelli e viaggi avventurosi."
+            "Highly detailed, painterly style with a rugged yet cinematic aesthetic. "
+            "Rich textures, earthy tones, and a palette dominated by dusty browns, deep reds, and sunset golds. "
+            "Inspired by classic Western art, frontier landscapes, and vintage posters. "
+            "Iconic elements like cowboy hats, revolvers, and stagecoaches evoke the romance of the Wild West. "
+            "Designed for a tabletop card game, ensuring clarity, readability, and an immersive sense of adventure."
         )
     },
     "Steampunk": {
         "nome": "Steampunk",
         "stile_immagine": (
-            "Un'estetica retrofuturistica che fonde l'epoca vittoriana con invenzioni meccaniche. "
-            "Abiti d'epoca, ingranaggi in ottone, macchinari complessi e ambientazioni industriali. "
-            "Colori caldi e metallici, texture invecchiate e dettagli elaborati che trasmettono un'atmosfera "
-            "di mistero e innovazione, perfetti per una carta di gioco che vuole raccontare storie fuori dal comune."
+            "Highly detailed, painterly style with a fusion of Victorian elegance and mechanical ingenuity. "
+            "Brass gears, intricate machinery, steam-powered contraptions, and industrial backdrops "
+            "create a retro-futuristic atmosphere. "
+            "A color palette dominated by warm metallics, deep browns, and aged parchment tones. "
+            "Inspired by classic steampunk literature and art, ensuring a visually immersive experience "
+            "for a tabletop card game with a sense of mystery and innovation."
         )
     },
     "Alieni": {
         "nome": "Alieni",
         "stile_immagine": (
-            "Un'estetica fantascientifica e surreale, con creature d'aspetto extraterrestre, "
-            "paesaggi cosmici e tecnologie aliene. Combina elementi di luce fluorescente, "
-            "forme organiche e architetture impossibili, creando un'atmosfera misteriosa e intrigante. "
-            "Ideale per un gioco di carte che trasporta i giocatori in un universo lontano e pieno di meraviglie."
+            "Highly detailed, painterly style with a futuristic and surreal aesthetic. "
+            "Glowing neon hues, organic and biomechanical forms, and impossible cosmic landscapes "
+            "blend to create an otherworldly atmosphere. "
+            "A color palette dominated by electric blues, deep purples, and iridescent greens. "
+            "Inspired by sci-fi concept art and space opera visuals, ensuring clarity, readability, "
+            "and a visually stunning experience for a tabletop card game set in a mysterious universe."
         )
     },
     "Homo Sapiens": {
         "nome": "Homo Sapiens",
         "stile_immagine": (
-            "Un'interpretazione moderna e riflessiva dell'evoluzione umana, con un mix di realismo e "
-            "elementi simbolici che rappresentano la storia e la cultura della nostra specie. "
-            "Utilizza toni naturali, texture ispirate alla materia organica e dettagli che richiamano "
-            "l'arte preistorica e contemporanea, per creare un'immagine che unisce il passato remoto al presente."
+            "Highly detailed, painterly style with a blend of realism and symbolic representation. "
+            "Natural tones, textures inspired by organic matter, and visual elements reflecting "
+            "human evolution, history, and cultural development. "
+            "A fusion of prehistoric cave paintings, Renaissance anatomical studies, and modern artistic interpretations. "
+            "Designed for a tabletop card game, ensuring clarity, readability, and a thought-provoking visual narrative "
+            "that bridges the ancient past with the present."
         )
-    },
+    }   
 }
 
 
@@ -137,12 +144,27 @@ class Character(BaseModel):
     intelligenza: int
     descrizione: str
     english_description:str
+    storia: str
 
 # Definizione dello schema per la risposta contenente una lista di personaggi
 class CharactersResponse(BaseModel):
     personaggi: list[Character]
 
-def generate_story(character: Character, nome_ambientazione, creativita): 
+class ActionCard(BaseModel):
+    nome_azione: str 
+    breve_descrizione: str
+    breve_descrizione_inglese: str
+
+class ActionCardsResponse(BaseModel):
+    carte_azione: list[ActionCard]
+
+def generate_story(character: Character, carte_azione: ActionCardsResponse, nome_ambientazione, creativita): 
+    descrizione_carte_azione = ""
+    descrizione_personaggi = ""
+    if carte_azione:
+        descrizione_carte_azione = f"Ecco anche le CARTE AZIONE del gioco: {carte_azione.model_dump_json()}"
+    if character:
+        descrizione_personaggi = f"Ecco i personaggi del gioco di RUOLO a carte.   {character.model_dump_json()}"
     response = clientOpenAI.chat.completions.create(
         model='gemini-2.0-flash-thinking-exp-01-21',
         n=1,
@@ -152,7 +174,7 @@ def generate_story(character: Character, nome_ambientazione, creativita):
             {"role": "system", "content": f"Tu sei un creatore di giochi di ruolo a CARTE. Crea un regolamento per un gioco di ruolo sul {nome_ambientazione} sulla base dei personaggi che ti fornirò"},
             {
                 "role": "user",
-                "content": f"Ecco i personaggi del gioco di RUOLO a carte. Crea un regolamento!!!  {character.model_dump_json()}"
+                "content": f"Crea un regolamento!!! {descrizione_personaggi} \n\n {descrizione_carte_azione}"
             }
         ]
     )
@@ -170,8 +192,9 @@ def generate_story(character: Character, nome_ambientazione, creativita):
 def generate_ai(num_personaggi, nome_ambientazione, creativita):
     # Costruzione del prompt in italiano per generare i personaggi
     prompt = (
-        f"Genera {num_personaggi} personaggi per un gioco di ruolo a carte ambientato nel {nome_ambientazione}. "
-        "Ogni personaggio deve avere i seguenti campi specificati nel modello. "
+        f"Genera {num_personaggi} personaggi AMBIENTATI NELL'EPOCA {nome_ambientazione},  per un gioco di ruolo a carte."
+        "Ogni personaggio deve avere i seguenti campi specificati nel modello ed devono essere COERENTI con l'epoca e l'ambientazione fornita (per esempio, se siamo nel Sacro Romano Impero devono essere nomi italiani) "
+        "nel campo STORIA, inventa la una BREVE storia di background del personaggio specificando aneddoti, e altre cose che rendono il personaggio unico. "
         "Restituisci il risultato in formato JSON seguendo lo schema fornito")
 
     # Esecuzione della chiamata all'API utilizzando il formato response_format
@@ -188,6 +211,24 @@ def generate_ai(num_personaggi, nome_ambientazione, creativita):
     print(characters_response)
     return characters_response
 
+def generate_card_ai(num_carte_azione, nome_ambientazione, creativita): 
+    prompt = (
+        f"Genera {num_carte_azione} CARTE AZIONE, per un gioco di ruolo ambientato in {nome_ambientazione}. "
+        "Attieniti allo schema formito e sii molto SINTETICO")
+
+    # Esecuzione della chiamata all'API utilizzando il formato response_format
+    completion = clientOpenAI.beta.chat.completions.parse(
+        model=MODEL,
+        messages=[
+            {"role": "system", "content": f"Sei un assistente utile per la generazione di CARTE AZIONE per un gioco di RUOLO sul {nome_ambientazione}."},
+            {"role": "user", "content": prompt},
+        ],
+        temperature=creativita,
+        response_format=ActionCardsResponse,
+    )
+    action_card_response = completion.choices[0].message.parsed
+    return action_card_response
+
 # Funzione per generare le immagini, con gestione errori e retry dopo 10 secondi
 def generate_image(prompt, max_retries=5):
     client = Together(api_key=api_together)
@@ -205,19 +246,33 @@ def generate_image(prompt, max_retries=5):
             )
             return response.data  # Una lista di oggetti con attributo b64_json
         except Exception as e:
-            st.error(f"Errore durante la generazione delle immagini: {e}. Riprovo tra 10 secondi...")
-            time.sleep(10)
+            print(f"Errore durante la generazione delle immagini: {e}. Riprovo tra 10 secondi...")
+            time.sleep(9)
             retries += 1
     st.error("Numero massimo di tentativi raggiunto. Impossibile generare le immagini.")
     return None
 
-def generate_images(character: Character, stile_immagine, num_immagini):
+def generate_images(character: Character, carta_azione: ActionCard, stile_immagine, num_immagini):
     # Lista per salvare le immagini generate come tuple (nome_file, bytes)
-    prompt = f"{character.english_description} {stile_immagine}. (INSERT TEXT NAME = {character.nome})"
-    images_bytes_list = []
-    if character.nome != "":
-        st.subheader(f"{character.nome} 🦸‍♂️")
-        st.markdown(f"- **Classe:** {character.classe}\n- **Forza:** {character.forza}\n- **Destrezza:** {character.destrezza}\n- **Intelligenza:** {character.intelligenza}\n- **Descrizione:** {character.descrizione}", unsafe_allow_html=True)   
+    if character:
+        prompt = f"{character.english_description} {stile_immagine}. (INSERT TEXT NAME = {character.nome})"
+        images_bytes_list = []
+        if character.nome != "":
+            st.subheader(f"{character.nome} 🦸‍♂️")
+            st.write(
+                f"- **Classe:** {character.classe}\n"
+                f"- **Forza:** {character.forza}\n"
+                f"- **Destrezza:** {character.destrezza}\n"
+                f"- **Intelligenza:** {character.intelligenza}\n"
+                f"- **Descrizione:** {character.descrizione}\n",
+                f"- **Storia:** {character.storia}"
+            )
+    if carta_azione:
+        prompt = f"PLAYING CARD FOR A CARD GAME WITHOUT CHARACTER WITH THIS BOLD INSCRIPTION IN THE CENTER OF THE CARD: '{carta_azione.nome_azione}'. USE THIS STYLE: {stile_immagine}.)"
+        images_bytes_list = []
+        st.subheader(f"{carta_azione.nome_azione} 🃏")
+        st.write(f"- **Descrizione:** {carta_azione.breve_descrizione}")        
+    print(prompt)
     for numero in range(num_immagini):
         images_data = generate_image(prompt)
         if images_data is not None:
@@ -233,7 +288,7 @@ def generate_images(character: Character, stile_immagine, num_immagini):
                     st.error(f"Errore nella visualizzazione dell'immagine {i+1}: {e}")
         else:
             st.error("Non è stato possibile generare le immagini. Riprova più tardi.")
-        time.sleep(5)
+        time.sleep(0.5)
     return images_bytes_list
 
 def main():
@@ -245,32 +300,50 @@ def main():
     stile_immagine = st.sidebar.text_area("Stile Immagine", stile_default, disabled=False)
     auto = False
     prompt_input = st.sidebar.text_input("Prompt Immagine (in inglese)", value="Soldier", disabled=auto)
-    auto = st.sidebar.toggle(label= 'Generazione automatica', value = True)
-    num_personaggi = st.sidebar.slider("Personaggi", min_value=1, max_value=30, value=5, disabled=not auto)
     num_immagini = st.sidebar.slider("Variazioni Immagini", min_value=1, max_value=6, value=2)
+    auto = st.sidebar.toggle(label= 'Generazione automatica', value = True)
+    num_personaggi = st.sidebar.slider("Personaggi", min_value=0, max_value=30, value=5, disabled=not auto)
+    num_carte_azione = st.sidebar.slider("Carte Azione", min_value=0, max_value=10, value=0, disabled=not auto)
+    genera_regolamento = st.sidebar.toggle(label= 'Regolamento', value = True, disabled=not auto)
     creativita = st.sidebar.slider("Creativita", min_value=0.1, max_value=1.0, value=0.8, step=0.1)
     submit_button = st.sidebar.button(label="Genera Immagine", type="primary", use_container_width=True)
-    st.write("Forgia il tuo **destino nell'Impero**: crea, combatti e domina con nel più grande gioco di ruolo a carte generato dall'AI")
+    st.write("Forgia il tuo **destino nell'Impero**: crea, combatti e domina  nel più grande gioco di ruolo a carte generato dall'AI")
     
     if submit_button:
         if not prompt_input.strip() and not auto:
             st.error("Per favore, inserisci un prompt per l'immagine!")
             return
+        carte_azione = None
+        characters = None
         if auto: 
-            with st.spinner('Generazione Personaggi'):
-                characters = generate_ai(num_personaggi, nome_ambientazione, creativita)
-                st.subheader('Personaggi 🎭')
-                df = pd.DataFrame([{k: v for k, v in character.model_dump().items() if k != "english_description"} for character in characters.personaggi])
-                st.dataframe(df, hide_index=True)                
-                st.divider()
-            generate_story(characters, nome_ambientazione, creativita)
-            st.divider()
+            if num_personaggi > 0: 
+                with st.spinner('Generazione Personaggi'):
+                    characters = generate_ai(num_personaggi, nome_ambientazione, creativita)
+                    st.subheader('Personaggi 🎭')
+                    df = pd.DataFrame([{k: v for k, v in character.model_dump().items() if k != "english_description"} for character in characters.personaggi])
+                    st.dataframe(df, hide_index=True, use_container_width=True)                
+                    st.divider()
+            if num_carte_azione > 0: 
+                with st.spinner('Generazione Carte Azione'):
+                    carte_azione = generate_card_ai(num_carte_azione, nome_ambientazione, creativita)
+                    st.subheader('Carte Azione 📒')
+                    df = pd.DataFrame([{k: v for k, v in carta_azione.model_dump().items() if k != "breve_descrizione_inglese"} for carta_azione in carte_azione.carte_azione])
+                    st.dataframe(df, hide_index=True, use_container_width=True)                
+                    st.divider()
+            if genera_regolamento:
+                with st.spinner('Generazione Regolamento'): 
+                    generate_story(characters, carte_azione, nome_ambientazione, creativita)
+                    st.divider()
         else:
-            characters = CharactersResponse(personaggi=[Character(nome="", classe="Guerriero", forza=10, destrezza=8, intelligenza=6, descrizione="Un forte guerriero", english_description=prompt_input)])
+            characters = CharactersResponse(personaggi=[Character(nome="", classe="Guerriero", forza=10, destrezza=8, intelligenza=6, storia="", descrizione="Un forte guerriero", english_description=prompt_input)])
         with st.spinner('Generazione Immagini'):
             images = []
-            for character in characters.personaggi:
-                images.extend(generate_images(character, stile_immagine, num_immagini))
+            if characters: 
+                for character in characters.personaggi:
+                    images.extend(generate_images(character, None, stile_immagine, num_immagini))
+            if carte_azione:
+                for  carta_azione in carte_azione.carte_azione:
+                    images.extend(generate_images(None, carta_azione, stile_immagine, num_immagini))
             if images:
                 zip_buffer = io.BytesIO()
                 with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file: