Update app.py

app.py

@@ -5,16 +5,12 @@ from PIL import Image
 import os
 import io
 import uuid
+import threading
+import time
 from dotenv import load_dotenv
 load_dotenv()
 
 # --- CONFIGURATION ---
-api_key = os.environ.get("GOOGLE_API_KEY")
-
-client = None
-if api_key:
-    client = genai.Client(api_key=api_key)
-
 MODELS = {
     "🧠 Gemini 3 Pro Preview (Recommended)": "gemini-3-pro-image-preview",
     "⚡ Gemini 2.5 Flash (Fast)": "gemini-2.5-flash-image"
@@ -28,11 +24,9 @@ TEMP_CHAT_DIR = "temp_chat_images"
 os.makedirs(TEMP_CHAT_DIR, exist_ok=True)
 
 # --- UTILS ---
-
-def get_client():
-    if not client:
-        raise gr.Error("API Key missing. Please set GOOGLE_API_KEY environment variable.")
-    return client
+def get_client(api_key):
+    if not api_key: raise gr.Error("API Key manquante")
+    return genai.Client(api_key=api_key)
 
 def safe_process_image(part):
     """Converts raw data to PIL Image."""
@@ -81,20 +75,15 @@ def process_response(response):
 
 def update_api_key(new_key):
     """Updates the global client with the user's API key."""
-    global client
+    # global client # NOT global for HF Spaces
     if not new_key:
-        return "⚠️ Please enter a valid API Key."
-    
-    try:
-        # Attempt to initialize the client
-        client = genai.Client(api_key=new_key)
-        return "✅ API Key configured successfully! You can now use the application."
-    except Exception as e:
-        return f"❌ Configuration Error: {str(e)}"
+        return "⚠️ Clé invalide", None # None ne changera pas l'état
 
-def generate_studio(prompt, model_ui, ratio, resolution, grounding):
+    return "✅ Clé enregistrée pour cette session !", new_key
+
+def generate_studio(prompt, model_ui, ratio, resolution, grounding, user_api_key):
     """Standard T2I Generation"""
-    cli = get_client()
+    cli = get_client(user_api_key)
     model_name = MODELS[model_ui]
     
     img_conf = {"aspect_ratio": ratio}
@@ -118,9 +107,9 @@ def generate_studio(prompt, model_ui, ratio, resolution, grounding):
     except Exception as e:
         raise gr.Error(f"API Error: {str(e)}")
 
-def generate_composition(prompt, files, model_ui, ratio, resolution):
+def generate_composition(prompt, files, model_ui, ratio, resolution, user_api_key):
     """Composition I2I"""
-    cli = get_client()
+    cli = get_client(user_api_key)
     model_name = MODELS[model_ui]
     
     if not files: raise gr.Error("No input images provided.")
@@ -157,8 +146,8 @@ def generate_composition(prompt, files, model_ui, ratio, resolution):
 
 # --- CHAT LOGIC ---
 
-def init_chat_session(model_ui, grounding):
-    cli = get_client()
+def init_chat_session(model_ui, grounding, user_api_key):
+    cli = get_client(user_api_key)
     model_name = MODELS[model_ui]
     
     tools = None
@@ -174,10 +163,10 @@ def init_chat_session(model_ui, grounding):
     )
     return chat
 
-def chat_respond(message, history, chat_state, img_input, model_ui, grounding):
+def chat_respond(message, history, chat_state, img_input, model_ui, grounding, user_api_key):
     """Iterative chat management with image history"""
     if chat_state is None:
-        chat_state = init_chat_session(model_ui, grounding)
+        chat_state = init_chat_session(model_ui, grounding, user_api_key)
     
     # --- 1. User message prep ---
     contents = [message]
@@ -238,8 +227,11 @@ def chat_respond(message, history, chat_state, img_input, model_ui, grounding):
         bot_err_obj = {"role": "assistant", "content": err_msg}
         return "", history + [user_message_obj, bot_err_obj], chat_state, []
 
-def clear_chat(model_ui, grounding):
-    new_chat = init_chat_session(model_ui, grounding)
+def clear_chat(model_ui, grounding, user_api_key):
+    # Si pas de clé, on retourne juste vide, sinon on recrée un chat
+    if not user_api_key:
+        return [], None, []
+    new_chat = init_chat_session(model_ui, grounding, user_api_key)
     return [], new_chat, []
 
 # --- GRADIO INTERFACE ---
@@ -256,6 +248,8 @@ with gr.Blocks(title="Nano Vision Studio") as demo:
     gr.Markdown("# Nano 🍌 Vision Studio")
     gr.Markdown("### The Ultimate Interface: 4K Generation, Grounding, Multi-Image Composition & Iterative Chat")
     
+    # État pour stocker la clé API de CET utilisateur spécifique
+    user_api_key_state = gr.State(os.environ.get("GOOGLE_API_KEY", ""))
     # Chat Session State
     chat_state = gr.State(None)
 
@@ -287,7 +281,7 @@ with gr.Blocks(title="Nano Vision Studio") as demo:
             api_btn.click(
                 update_api_key,
                 inputs=[api_input],
-                outputs=[api_status]
+                outputs=[api_status, user_api_key_state]
             )
         
         # --- TAB 1 : CREATION STUDIO ---
@@ -319,7 +313,7 @@ with gr.Blocks(title="Nano Vision Studio") as demo:
 
             t1_btn.click(
                 generate_studio,
-                inputs=[t1_prompt, t1_model, t1_ratio, t1_res, t1_grounding],
+                inputs=[t1_prompt, t1_model, t1_ratio, t1_res, t1_grounding, user_api_key_state],
                 outputs=[t1_gallery, t1_text, t1_thought_imgs, t1_thought_txt]
             )
 
@@ -344,7 +338,7 @@ with gr.Blocks(title="Nano Vision Studio") as demo:
 
             t2_btn.click(
                 generate_composition,
-                inputs=[t2_prompt, t2_files, t2_model, t2_ratio, t2_res],
+                inputs=[t2_prompt, t2_files, t2_model, t2_ratio, t2_res, user_api_key_state],
                 outputs=[t2_gallery, t2_text]
             )
             
@@ -375,13 +369,13 @@ with gr.Blocks(title="Nano Vision Studio") as demo:
 
             chat_btn.click(
                 chat_respond,
-                inputs=[chat_input, chat_history, chat_state, chat_img, c_model, c_grounding],
+                inputs=[chat_input, chat_history, chat_state, chat_img, c_model, c_grounding, user_api_key_state],
                 outputs=[chat_input, chat_history, chat_state, chat_gallery_zoom]
             )
             
             clear_btn.click(
                 clear_chat,
-                inputs=[c_model, c_grounding],
+                inputs=[c_model, c_grounding, user_api_key_state],
                 outputs=[chat_history, chat_state, chat_gallery_zoom]
             )
             
@@ -411,7 +405,34 @@ with gr.Blocks(title="Nano Vision Studio") as demo:
             - **Resolution**: Use the "Pro" model to unlock 4K output.
             """)
 
+def cleanup_old_files():
+    """Supprime les fichiers vieux de plus de 1h toutes les 10 minutes."""
+    while True:
+        try:
+            now = time.time()
+            # 3600 secondes = 1 heure
+            cutoff = now - 3600  
+            
+            if os.path.exists(TEMP_CHAT_DIR):
+                for filename in os.listdir(TEMP_CHAT_DIR):
+                    filepath = os.path.join(TEMP_CHAT_DIR, filename)
+                    if os.path.isfile(filepath):
+                        # Si le fichier est plus vieux que le cutoff
+                        if os.path.getmtime(filepath) < cutoff:
+                            try:
+                                os.remove(filepath)
+                                print(f"🧹 Supprimé : {filename}")
+                            except Exception:
+                                pass # On ignore les erreurs ponctuelles
+        except Exception as e:
+            print(f"⚠️ Erreur worker : {e}")
+        
+        # Pause de 30 minutes entre chaque vérification
+        time.sleep(1800)
+
 if __name__ == "__main__":
+    # Démarrage du nettoyage en arrière-plan (daemon=True pour qu'il s'arrête quand l'app quitte)
+    threading.Thread(target=cleanup_old_files, daemon=True).start()
     # Configuration Gradio 6.0
     demo.queue(default_concurrency_limit=20)
     
@@ -420,5 +441,8 @@ if __name__ == "__main__":
         theme=gr.themes.Soft(), 
         css=css,
         max_threads=40,
-        show_error=True
+        show_error=True,
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
     )