Update app.py

app.py

@@ -21,6 +21,16 @@ clientOpenAI = OpenAI(
 )
 
 ambientazioni = {
+    "Roma Repubblicana": {
+        "nome": "Roma Repubblicana",
+        "stile_immagine": (
+            "A painterly, highly detailed visual narrative rich with graphic elements Roman Repubblic "
+            "Incorporate ornate Roman columns, laurel wreaths, engraved inscriptions, and intricate mosaic patterns. "
+            "Textures of weathered marble, aged frescoes, and tattered papyrus blend with a dramatic chiaroscuro effect. "
+            "A color palette of imperial red, bronze, and muted gold accentuates secretive senatorial silhouettes and subtle sigils. "
+            "Designed for a tabletop card game, this style captures clandestine intrigue and political tension with dynamic visual flair."
+        )
+    },
     "Sacro Romano Impero": {
         "nome": "Sacro Romano Impero",
         "stile_immagine": (
@@ -252,12 +262,21 @@ def generate_image(prompt, max_retries=5):
     st.error("Numero massimo di tentativi raggiunto. Impossibile generare le immagini.")
     return None
 
-def generate_images(character: Character, carta_azione: ActionCard, stile_immagine, num_immagini):
+def generate_images(character: Character, carta_azione: ActionCard, stile_immagine, num_immagini, tipo_carta: str = None, nome_carta: str = None, colore_bordo: str = None):
     # Lista per salvare le immagini generate come tuple (nome_file, bytes)
     if character:
-        prompt = f"{character.english_description} {stile_immagine}. (INSERT TEXT NAME = {character.nome})"
         images_bytes_list = []
-        if character.nome != "":
+        if character.nome == "":
+            if tipo_carta == "Personaggio":
+                prompt = f"{character.english_description} WITH THIS BOLD INSCRIPTION '{nome_carta}'. BORDER COLOR = {colore_bordo}). Use this Style: {stile_immagine}"
+            if tipo_carta == "Fronte":
+                prompt = f"FRONT OF CARD GAME (WITH THIS BOLD INSCRIPTION IN THE CENTER OF THE CARD = '{nome_carta}'. BORDER COLOR = {colore_bordo}). Use this Syle: {stile_immagine}"
+            if tipo_carta == "Retro":
+                prompt = ("Design an ornate playing card back with symmetrical filigree"
+                          "At the center, prominently display the name '" + nome_carta + "' in an elegant, calligraphic style. ")                
+                prompt += f"BACKGROUND COLOR = {colore_bordo}. Use this Syle: {stile_immagine}"
+        else:
+            prompt = f"{character.english_description} {stile_immagine}. (INSERT TEXT NAME = {character.nome})"
             st.subheader(f"{character.nome} 🦸‍♂️")
             st.write(
                 f"- **Classe:** {character.classe}\n"
@@ -299,20 +318,28 @@ def main():
     nome_ambientazione = ambientazioni[selected_ambientazione]["nome"]
     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)
-    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)
+    prompt_input = ""
+    if auto == False:
+        tipo_carta = st.sidebar.selectbox("Tipo Carta",("Personaggio", "Fronte", "Retro"))
+        if tipo_carta == "Personaggio":
+            prompt_input = st.sidebar.text_input("Prompt Immagine (in inglese)", value="Soldier")
+        nome_carta = st.sidebar.text_input("Nome Carta", value="Carmine")
+        colore_bordo = st.sidebar.selectbox("Colore Bordo",("Blue", "Red", "Yellow", "Green", "Pink", "Black", "White"))
+    else:
+        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)
+        tipo_carta = None
+        nome_carta = None
+        colore_bordo = None
+
+    num_immagini = st.sidebar.slider("Variazioni Immagini", min_value=1, max_value=6, value=2)
     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  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: 
@@ -340,7 +367,7 @@ def main():
             images = []
             if characters: 
                 for character in characters.personaggi:
-                    images.extend(generate_images(character, None, stile_immagine, num_immagini))
+                    images.extend(generate_images(character, None, stile_immagine, num_immagini, tipo_carta, nome_carta, colore_bordo))
             if carte_azione:
                 for  carta_azione in carte_azione.carte_azione:
                     images.extend(generate_images(None, carta_azione, stile_immagine, num_immagini))
@@ -361,4 +388,4 @@ def main():
 
 if __name__ == "__main__":
     st.set_page_config(page_title="Imperium AI", page_icon="🏰", layout="wide")
-    main()
+    main()