Spaces:
Running
Running
| import streamlit as st | |
| import time | |
| import base64 | |
| import io | |
| import zipfile | |
| from PIL import Image | |
| from together import Together | |
| import os | |
| from dotenv import load_dotenv | |
| from pydantic import BaseModel | |
| from openai import OpenAI | |
| import pandas as pd | |
| load_dotenv() | |
| api_together = os.getenv("TOGETHER_API_KEY") | |
| api_gemini = os.getenv("API_GEMINI") | |
| MODEL = "gemini-2.0-flash-exp" | |
| clientOpenAI = OpenAI( | |
| api_key=api_gemini, | |
| base_url="https://generativelanguage.googleapis.com/v1beta/openai/" | |
| ) | |
| LOGO_STYLES = { | |
| "Minimalist": { | |
| "nome": "Minimalist", | |
| "stile_immagine": ( | |
| "A minimalist logo design prompt focusing on simplicity and essential elements. " | |
| "Clean lines, a limited color palette, and strategic use of negative space. " | |
| "Ideal for conveying modernity and clarity." | |
| ) | |
| }, | |
| "Modern": { | |
| "nome": "Modern", | |
| "stile_immagine": ( | |
| "A modern logo design prompt featuring abstract forms, innovative color combinations, " | |
| "and dynamic geometric shapes. Sleek typography and a contemporary aesthetic make it perfect for tech companies and startups." | |
| ) | |
| }, | |
| "Retro": { | |
| "nome": "Retro", | |
| "stile_immagine": ( | |
| "A retro logo design prompt evoking nostalgia with vintage color schemes, classic typography, " | |
| "and design elements reminiscent of the past. Perfect for brands that want to express heritage and authenticity." | |
| ) | |
| }, | |
| "Vintage": { | |
| "nome": "Vintage", | |
| "stile_immagine": ( | |
| "A vintage logo design prompt inspired by bygone eras. Emphasizes handcrafted details, worn textures, " | |
| "and a nostalgic atmosphere, ideal for artisanal products or brands with a long-standing tradition." | |
| ) | |
| }, | |
| "Geometric": { | |
| "nome": "Geometric", | |
| "stile_immagine": ( | |
| "A geometric logo design prompt that leverages simple, precise shapes, clean lines, and symmetry. " | |
| "Communicates order, professionalism, and a rational approach to design." | |
| ) | |
| }, | |
| "Typographic": { | |
| "nome": "Typographic", | |
| "stile_immagine": ( | |
| "A typographic logo design prompt focused on the creative use of lettering. " | |
| "Bold typography paired with minimal color usage highlights the strength of word-based identities." | |
| ) | |
| }, | |
| } | |
| class Logo(BaseModel): | |
| nome: str | |
| descrizione: str | |
| english_description:str | |
| class Loghi(BaseModel): | |
| loghi: list[Logo] | |
| def generate_ai(num_loghi, tema, creativita): | |
| prompt = ( | |
| f"Genera {num_loghi} prompt per la generazione immagini che trasmetta questo: {tema}" | |
| "Sii molto SINTETICO e usa SIMBOLI stilizzati e non troppi oggetti. Restituisci il risultato in formato JSON seguendo lo schema fornito") | |
| completion = clientOpenAI.beta.chat.completions.parse( | |
| model=MODEL, | |
| messages=[ | |
| {"role": "system", "content": f"Sei un assistente utile per la generazione di IDEE per la generazioni immagini su questo tema: {tema}."}, | |
| {"role": "user", "content": prompt}, | |
| ], | |
| temperature=creativita, | |
| response_format=Loghi, | |
| ) | |
| loghi = completion.choices[0].message.parsed | |
| print(loghi) | |
| return loghi | |
| # 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) | |
| retries = 0 | |
| while retries < max_retries: | |
| try: | |
| response = client.images.generate( | |
| prompt=prompt, | |
| model="black-forest-labs/FLUX.1-schnell-Free", | |
| width=1024, | |
| height=1024, | |
| steps=4, | |
| n=1, | |
| response_format="b64_json" | |
| ) | |
| return response.data # Una lista di oggetti con attributo b64_json | |
| except Exception as e: | |
| 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(logo: Logo, nome_stile, stile_immagine, num_immagini, colori: list[str] = None): | |
| if logo: | |
| images_bytes_list = [] | |
| colors_str = " ".join(colori) if colori else "" | |
| prompt = f"Create a Simple Logo in {nome_stile} STYLE background white, and COLORS '{colors_str}' of '{logo.english_description}'. use this style {stile_immagine}" | |
| st.subheader(f"{logo.nome} 🖌️") | |
| st.write(logo.descrizione) | |
| print(prompt) | |
| for numero in range(num_immagini): | |
| images_data = generate_image(prompt) | |
| if images_data is not None: | |
| for i, img_obj in enumerate(images_data): | |
| try: | |
| image_bytes = base64.b64decode(img_obj.b64_json) | |
| image = Image.open(io.BytesIO(image_bytes)) | |
| st.image(image, caption="") | |
| img_byte_arr = io.BytesIO() | |
| image.save(img_byte_arr, format='PNG') | |
| images_bytes_list.append((f"image_{numero+1}_{i+1}.png", img_byte_arr.getvalue())) | |
| except Exception as e: | |
| 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(0.5) | |
| return images_bytes_list | |
| def main(): | |
| st.title("Logo Generator AI 🎨") | |
| st.sidebar.header("Impostazioni") | |
| selected_stile = st.sidebar.selectbox("Ambientazione", list(LOGO_STYLES.keys()), index=0) | |
| stile_default = LOGO_STYLES[selected_stile]["stile_immagine"] | |
| nome_stile = LOGO_STYLES[selected_stile]["nome"] | |
| stile_immagine = st.sidebar.text_area("Stile Immagine", stile_default, disabled=False) | |
| auto = True | |
| tema = st.sidebar.text_input("Tema Logo", value="Formazione Aziendale") | |
| #auto = st.sidebar.toggle(label= 'Generazione automatica', value = True) | |
| prompt_input = "" | |
| colori = st.sidebar.multiselect("Colori", ["Blue", "Orange", "Green", "Yellow", "Red", "Purple"], ["Blue", "Orange"]) | |
| num_loghi = st.sidebar.slider("Loghi", min_value=0, max_value=30, value=10, disabled=not auto) | |
| num_immagini = st.sidebar.slider("Variazioni", min_value=1, max_value=6, value=2) | |
| creativita = st.sidebar.slider("Creativita", min_value=0.1, max_value=1.0, value=0.95, step=0.1) | |
| submit_button = st.sidebar.button(label="Genera Immagine", type="primary", use_container_width=True) | |
| st.write("Genera il tuo **Logo Aziendale** tramite l'AI") | |
| if submit_button: | |
| if auto: | |
| if num_loghi > 0: | |
| with st.spinner('Generazione Loghi'): | |
| loghi = generate_ai(num_loghi, tema, creativita) | |
| st.subheader('Loghi 💡') | |
| df = pd.DataFrame([{k: v for k, v in logo.model_dump().items() if k != ""} for logo in loghi.loghi]) | |
| st.dataframe(df, hide_index=True, use_container_width=True) | |
| st.divider() | |
| with st.spinner('Generazione Immagini'): | |
| images = [] | |
| if loghi: | |
| for logo in loghi.loghi: | |
| images.extend(generate_images(logo, nome_stile, stile_immagine, num_immagini, colori)) | |
| if images: | |
| zip_buffer = io.BytesIO() | |
| with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file: | |
| for file_name, file_bytes in images: | |
| zip_file.writestr(file_name, file_bytes) | |
| zip_buffer.seek(0) | |
| st.download_button( | |
| label="Download All Images", | |
| data=zip_buffer, | |
| file_name="images.zip", | |
| mime="application/zip", | |
| type='primary' | |
| ) | |
| st.success("Immagini generate con successo!") | |
| if __name__ == "__main__": | |
| st.set_page_config(page_title="Logo Generator AI", page_icon="🎨", layout="wide") | |
| main() |