Upload 2 files

app.py

@@ -1,16 +1,44 @@
 import gradio as gr
 import os
 import time
+import requests
+import base64
+import jwt
+import io
+from PIL import Image
+
+# Импортируем либу из твоего рабочего кода
+try:
+    from google import genai
+    from google.genai import types
+except ImportError:
+    pass
 
 # --- SECRETS AND CONFIGURATION ---
-NANOBANANA_API_KEY = os.getenv("NANOBANANA_API_KEY")
-KLING_API_KEY = os.getenv("KLING_API_KEY")
+# Используем ключ Google из твоего кода (добавь GOOGLE_API_KEY в Secrets)
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY") 
+KLING_ACCESS_KEY = os.getenv("KLING_ACCESS_KEY")
+KLING_SECRET_KEY = os.getenv("KLING_SECRET_KEY")
 
 SYSTEM_PROMPT = """Hyper-realistic Instagram influencer lifestyle photo, shot on iPhone 15 Pro. The exact person [refer to subject images] is looking directly at the camera with a warm, highly attractive, and confident expression. Flawless, beautiful, glowing natural skin with a soft aesthetic finish (no harsh shadows, no rough textures). They are elegantly yet casually holding the product [refer to product images] close to the camera, showing it to the viewers. Flattering soft daylight from a window combined with a subtle ring-light glow on the face. Bright, aesthetic, cozy modern room background. High-end social media blogger vibe, perfect facial likeness, highly detailed but soft and visually pleasing."""
 
+# --- JWT TOKEN GENERATOR FOR KLING ---
+def generate_kling_token(ak, sk):
+    """Generates a short-lived JWT token required by Kling API"""
+    headers = {
+        "alg": "HS256",
+        "typ": "JWT"
+    }
+    payload = {
+        "iss": ak,
+        "exp": int(time.time()) + 1800,
+        "nbf": int(time.time()) - 5
+    }
+    return jwt.encode(payload, sk, algorithm="HS256", headers=headers)
+
+
 def create_photo(blogger_images, product_images):
     logs = ""
-    # Check if images are uploaded
     if not blogger_images or not product_images:
         raise gr.Error("Please upload photos of both the blogger and the product!")
     
@@ -18,119 +46,192 @@ def create_photo(blogger_images, product_images):
     yield logs, None
     time.sleep(1)
     
-    logs += "[Storage] Uploading blogger and product images to the cloud...\n"
-    yield logs, None
-    time.sleep(1.5)
-    
-    logs += f"[NanoBanana Pro] Sending API request. Prompt: {SYSTEM_PROMPT[:50]}...\n"
-    yield logs, None
-    time.sleep(1.5)
-    
-    logs += "[NanoBanana Pro] Generating image (applying style and likeness)...\n"
-    yield logs, None
-    time.sleep(2)
-    
-    logs += "[Success] Photo generated! Loading into interface...\n"
-    # Placeholder realistic photo for demo purposes
-    photo_url = "https://images.unsplash.com/photo-1611162617474-5b21e879e113?q=80&w=1000&auto=format&fit=crop"
-    
-    yield logs, photo_url
+    # =========================================================================
+    # DEMO FALLBACK (Если ключа нет)
+    # =========================================================================
+    if not GOOGLE_API_KEY:
+        logs += "[Warning] GOOGLE_API_KEY is not set! Running in Demo Fallback mode...\n"
+        yield logs, None
+        time.sleep(2)
+        logs += "[Success] Photo generated! Loading into interface...\n"
+        yield logs, "https://images.unsplash.com/photo-1611162617474-5b21e879e113?q=80&w=1000&auto=format&fit=crop"
+        return
+
+    # =========================================================================
+    # РЕАЛЬНЫЙ ВЫЗОВ (Из твоего кода)
+    # =========================================================================
+    try:
+        logs += "[Storage] Preparing blogger and product images...\n"
+        yield logs, None
+        
+        client = genai.Client(api_key=GOOGLE_API_KEY)
+        
+        # Собираем контент так же, как в твоем скрипте
+        contents = [SYSTEM_PROMPT]
+        
+        for img in blogger_images:
+            path = img if isinstance(img, str) else img.name
+            contents.append(Image.open(path))
+            
+        for img in product_images:
+            path = img if isinstance(img, str) else img.name
+            contents.append(Image.open(path))
+            
+        logs += f"[Google GenAI] Sending API request (gemini-3-pro-image-preview)...\n"
+        yield logs, None
+        
+        # Вызов генерации 1-в-1 как в твоем скрипте
+        resp = client.models.generate_content(
+            model="gemini-3-pro-image-preview",
+            contents=contents,
+            config=types.GenerateContentConfig(
+                response_modalities=['TEXT', 'IMAGE'],
+                image_config=types.ImageConfig(aspect_ratio="3:4", image_size="2K") # 3:4 идеально для блогера
+            )
+        )
+        
+        # Извлечение байтов картинки
+        img_data = None
+        for part in resp.parts:
+            if part.inline_data: 
+                img_data = part.inline_data.data
+                break
+                
+        if img_data:
+            logs += "[Success] Photo generated successfully! Loading to UI...\n"
+            # Превращаем байты в картинку для Gradio
+            result_img = Image.open(io.BytesIO(img_data))
+            yield logs, result_img
+        else:
+            logs += "[Error] API succeeded but returned no image data.\n"
+            yield logs, None
+            
+    except Exception as e:
+        err = str(e)
+        if "429" in err:
+            logs += f"[Error] Rate limited by API (429)\n"
+        else:
+            logs += f"[Exception] API Error: {err[:100]}\n"
+        yield logs, None
+
 
 def create_video(blogger_images, product_images):
     logs = ""
-    # Check if images are uploaded
     if not blogger_images or not product_images:
         raise gr.Error("Please upload photos of both the blogger and the product!")
 
-    logs += "[System] Initializing video generation...\n"
+    logs += "[System] Initializing Kling AI video generation...\n"
     yield logs, None
     time.sleep(1)
-    
-    logs += "[Kling API] POST /v1/videos/image2video -> Sending data...\n"
-    yield logs, None
-    time.sleep(1.5)
-    
-    logs += "[Kling API] Task ID: kl_9876543210ab. Status: QUEUED...\n"
-    yield logs, None
-    time.sleep(2)
-    
-    logs += "[Kling API] Status: PROCESSING (Rendering frames)...\n"
-    yield logs, None
-    time.sleep(2.5)
-    
-    logs += "[Success] Status: SUCCEED. Video is ready!\n"
-    # Placeholder video for demo purposes
-    video_url = "https://www.w3schools.com/html/mov_bbb.mp4"
-    
-    yield logs, video_url
 
-# --- CUSTOM UI STYLES ---
-custom_css = """
-.container { max-width: 1100px; margin: auto; }
-.output-media { border-radius: 16px; box-shadow: 0 10px 25px rgba(0,0,0,0.1); }
-button.primary { background: linear-gradient(90deg, #6366f1, #a855f7); border: none; }
-button.primary:hover { transform: translateY(-2px); box-shadow: 0 5px 15px rgba(168, 85, 247, 0.4); transition: all 0.3s ease; }
-.log-window textarea { 
-    background-color: #1e1e1e !important; 
-    color: #4ade80 !important; 
-    font-family: 'Courier New', Courier, monospace !important; 
-    font-size: 13px !important;
-}
-"""
+    if not KLING_ACCESS_KEY or not KLING_SECRET_KEY:
+        logs += "[Warning] Kling API Keys are not set! Running in Demo Fallback mode...\n"
+        yield logs, None
+        time.sleep(1)
+        logs += "[Kling API] POST /v1/videos/image2video -> Sending data...\n"
+        yield logs, None
+        time.sleep(1.5)
+        logs += "[Kling API] Task ID: kl_demo_98765. Status: SUBMITTED...\n"
+        yield logs, None
+        time.sleep(2)
+        logs += "[Kling API] Status: PROCESSING...\n"
+        yield logs, None
+        time.sleep(2.5)
+        logs += "[Success] Status: SUCCEED. Video is ready!\n"
+        yield logs, "https://www.w3schools.com/html/mov_bbb.mp4"
+        return
 
-# App Layout setup
-with gr.Blocks() as demo:
-    gr.Markdown("<h1 style='text-align: center;'>✨ AI Content Creator Pro</h1>")
-    gr.Markdown("<p style='text-align: center; color: gray;'>Hyper-realistic UGC Generation: NanoBanana Pro (Photo) & Kling AI (Video)</p>")
-    
-    with gr.Row():
-        # LEFT COLUMN (Inputs)
-        with gr.Column(scale=1):
-            gr.Markdown("### 📥 Inputs")
+    try:
+        logs += "[Security] Generating 30-min JWT Token using Access & Secret Keys...\n"
+        yield logs, None
+        api_token = generate_kling_token(KLING_ACCESS_KEY, KLING_SECRET_KEY)
+        
+        first_img = blogger_images[0]
+        img_path = first_img if isinstance(first_img, str) else first_img.name
+        
+        with open(img_path, "rb") as f:
+            base64_image = base64.b64encode(f.read()).decode('utf-8')
             
-            with gr.Group():
-                blogger_imgs = gr.File(label="📸 Blogger (1-5 photos)", file_count="multiple", file_types=["image"])
-                product_imgs = gr.File(label="🛍️ Product (1-5 photos)", file_count="multiple", file_types=["image"])
+        logs += "[Storage] Image converted to Base64 successfully.\n"
+        yield logs, None
+        
+        headers = {
+            "Authorization": f"Bearer {api_token}",
+            "Content-Type": "application/json"
+        }
+        
+        payload = {
+            "model_name": "kling-v2-6",
+            "image": f"data:image/jpeg;base64,{base64_image}",
+            "prompt": "Camera zooms out, the blogger smiles and enthusiastically shows the product directly to the camera, photorealistic, cinematic lighting",
+            "negative_prompt": "blur, distort, low quality, bad anatomy",
+            "duration": "5",
+            "mode": "pro",
+            "sound": "off"
+        }
+
+        logs += f"[Kling API] POST /v1/videos/image2video -> Starting task...\n"
+        yield logs, None
+        
+        post_url = "https://api-singapore.klingai.com/v1/videos/image2video"
+        response = requests.post(post_url, headers=headers, json=payload)
+        resp_json = response.json()
+        
+        if response.status_code != 200 or resp_json.get("code") != 0:
+            error_msg = resp_json.get('message', 'Unknown error')
+            logs += f"[Error] API Creation Failed: {error_msg}\n"
+            yield logs, None
+            return
             
-            with gr.Row():
-                btn_photo = gr.Button("Create Photo", variant="primary", size="lg")
-                btn_video = gr.Button("Create Video", variant="primary", size="lg")
-                
-            # TERMINAL LOGS
-            logs_output = gr.Textbox(
-                label="🖥️ System Logs", 
-                lines=6, 
-                interactive=False, 
-                elem_classes="log-window",
-                placeholder="Generation logs will appear here..."
-            )
-                
-        # RIGHT COLUMN (Outputs)
-        with gr.Column(scale=1):
-            gr.Markdown("### 📤 Output")
+        task_id = resp_json.get("data", {}).get("task_id")
+        logs += f"[Kling API] Task ID: {task_id}. Status: SUBMITTED...\n"
+        yield logs, None
+        
+        get_url = f"https://api-singapore.klingai.com/v1/videos/image2video/{task_id}"
+        
+        while True:
+            time.sleep(5) 
+            poll_resp = requests.get(get_url, headers=headers)
+            poll_json = poll_resp.json()
             
-            with gr.Tabs():
-                with gr.Tab("Photo"):
-                    out_photo = gr.Image(label="Photo Result", elem_classes="output-media")
-                with gr.Tab("Video"):
-                    out_video = gr.Video(label="Video Result", elem_classes="output-media")
+            if poll_json.get("code") != 0:
+                logs += f"[Error] API Polling Failed: {poll_json.get('message')}\n"
+                yield logs, None
+                break
+                
+            data = poll_json.get("data", {})
+            task_status = data.get("task_status", "").lower()
             
-    # Button triggers
-    btn_photo.click(
-        fn=create_photo, 
-        inputs=[blogger_imgs, product_imgs], 
-        outputs=[logs_output, out_photo]
-    )
-    
-    btn_video.click(
-        fn=create_video, 
-        inputs=[blogger_imgs, product_imgs], 
-        outputs=[logs_output, out_video]
-    )
+            if task_status == "succeed":
+                task_result = data.get("task_result", {})
+                videos = task_result.get("videos", [])
+                
+                if videos and "url" in videos[0]:
+                    video_url = videos[0]["url"]
+                    logs += "[Success] Status: SUCCEED. Video generation finished!\n"
+                    yield logs, video_url
+                else:
+                    logs += "[Error] Status SUCCEED, but no video URL found.\n"
+                    yield logs, None
+                break
+                
+            elif task_status == "failed":
+                fail_msg = data.get("task_status_msg", "Unknown reason")
+                logs += f"[Error] Task FAILED. Reason: {fail_msg}\n"
+                yield logs, None
+                break
+                
+            else:
+                logs += f"[Kling API] Status: {task_status.upper()}... Waiting 5s...\n"
+                yield logs, None
+                
+    except Exception as e:
+        logs += f"[Exception] Internal error occurred: {str(e)}\n"
+        yield logs, None
 
-if __name__ == "__main__":
-    # Launching the app with custom theme and CSS
-    demo.launch(
-        theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="violet"), 
-        css=custom_css
-    )
+# --- CUSTOM UI STYLES ---
+custom_css = """
+.container { max-width: 1100px; margin: auto; }
+.output-media { border-radius: 16px; box-shadow: 0 10px 25px rgba(0,0,0,0.1); }
+button.primary { background: linear-gradient(90deg, #6366f1, #a855f7); border: none; }
+button.primary:hover

requirements.txt

@@ -1,3 +1,4 @@
 gradio>=4.0.0
 requests
+PyJWT
 fal-client