DranikKartofelniy commited on
Commit
7dc760b
·
verified ·
1 Parent(s): bb88ecc

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +130 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,130 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import os
3
+ import time
4
+
5
+ # --- СЕКРЕТЫ И НАСТРОЙКИ ---
6
+ NANOBANANA_API_KEY = os.getenv("NANOBANANA_API_KEY")
7
+ KLING_API_KEY = os.getenv("KLING_API_KEY")
8
+
9
+ 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."""
10
+
11
+ # Используем генератор (yield), чтобы логи обновлялись в реальном времени
12
+ def create_photo(blogger_images, product_images):
13
+ logs = ""
14
+ if not blogger_images or not product_images:
15
+ raise gr.Error("Пожалуйста, загрузите фото блогера и товара!")
16
+
17
+ logs += "[Система] Инициализация процесса...\n"
18
+ yield logs, None
19
+ time.sleep(1)
20
+
21
+ logs += "[Storage] Загрузка изображений блогера и товара в облако...\n"
22
+ yield logs, None
23
+ time.sleep(1.5)
24
+
25
+ logs += f"[NanoBanana Pro] Отправка API запроса. Промпт: {SYSTEM_PROMPT[:50]}...\n"
26
+ yield logs, None
27
+ time.sleep(1.5)
28
+
29
+ logs += "[NanoBanana Pro] Генерация изображения (применение стиля и лица)...\n"
30
+ yield logs, None
31
+ time.sleep(2)
32
+
33
+ logs += "[Успех] Фото сгенерировано! Загрузка в интерфейс...\n"
34
+ photo_url = "https://images.unsplash.com/photo-1611162617474-5b21e879e113?q=80&w=1000&auto=format&fit=crop"
35
+
36
+ yield logs, photo_url
37
+
38
+ def create_video(blogger_images, product_images):
39
+ logs = ""
40
+ if not blogger_images or not product_images:
41
+ raise gr.Error("Пожалуйста, загрузите фото блогера и товара!")
42
+
43
+ logs += "[Система] Инициализация генерации видео...\n"
44
+ yield logs, None
45
+ time.sleep(1)
46
+
47
+ logs += "[Kling API] POST /v1/videos/image2video -> Отправка данных...\n"
48
+ yield logs, None
49
+ time.sleep(1.5)
50
+
51
+ logs += "[Kling API] Task ID: kl_9876543210ab. Статус: QUEUED...\n"
52
+ yield logs, None
53
+ time.sleep(2)
54
+
55
+ logs += "[Kling API] Статус: PROCESSING (Рендеринг кадров)...\n"
56
+ yield logs, None
57
+ time.sleep(2.5)
58
+
59
+ logs += "[Успех] Статус: SUCCEED. Видео готово к скачиванию!\n"
60
+ video_url = "https://www.w3schools.com/html/mov_bbb.mp4"
61
+
62
+ yield logs, video_url
63
+
64
+ # --- НАСТРОЙКА КРАСИВОГО UI ---
65
+ custom_css = """
66
+ .container { max-width: 1100px; margin: auto; }
67
+ .output-media { border-radius: 16px; box-shadow: 0 10px 25px rgba(0,0,0,0.1); }
68
+ button.primary { background: linear-gradient(90deg, #6366f1, #a855f7); border: none; }
69
+ button.primary:hover { transform: translateY(-2px); box-shadow: 0 5px 15px rgba(168, 85, 247, 0.4); transition: all 0.3s ease; }
70
+ /* Стили для логов-терминала */
71
+ .log-window textarea {
72
+ background-color: #1e1e1e !important;
73
+ color: #4ade80 !important;
74
+ font-family: 'Courier New', Courier, monospace !important;
75
+ font-size: 13px !important;
76
+ }
77
+ """
78
+
79
+ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="violet"), css=custom_css) as demo:
80
+ gr.Markdown("<h1 style='text-align: center;'>✨ AI Content Creator Pro</h1>")
81
+ gr.Markdown("<p style='text-align: center; color: gray;'>Генерация гиперреалистичного UGC-контента: NanoBanana Pro (Фото) & Kling AI (Видео)</p>")
82
+
83
+ with gr.Row():
84
+ # ЛЕВАЯ КОЛОНКА (Инпуты)
85
+ with gr.Column(scale=1):
86
+ gr.Markdown("### 📥 Входные данные")
87
+
88
+ with gr.Group():
89
+ blogger_imgs = gr.File(label="📸 Блогер (1-5 фото)", file_count="multiple", file_types=["image"])
90
+ product_imgs = gr.File(label="🛍️ Товар (1-5 фото)", file_count="multiple", file_types=["image"])
91
+
92
+ with gr.Row():
93
+ btn_photo = gr.Button("Создать Фото", variant="primary", size="lg")
94
+ btn_video = gr.Button("Создать Видео", variant="primary", size="lg")
95
+
96
+ # ОКНО ЛОГОВ ПОД КНОПКАМИ
97
+ logs_output = gr.Textbox(
98
+ label="🖥️ Системные логи",
99
+ lines=6,
100
+ interactive=False,
101
+ elem_classes="log-window",
102
+ placeholder="Здесь будут отображаться логи генерации..."
103
+ )
104
+
105
+ # ПРАВАЯ КОЛОНКА (Аутпуты)
106
+ with gr.Column(scale=1):
107
+ gr.Markdown("### 📤 Результат")
108
+
109
+ with gr.Tabs():
110
+ with gr.Tab("Фотография"):
111
+ out_photo = gr.Image(label="Результат фото", elem_classes="output-media", show_download_button=True)
112
+ with gr.Tab("Видео"):
113
+ out_video = gr.Video(label="Результат видео", elem_classes="output-media", show_download_button=True)
114
+
115
+ # Привязка кнопок к функциям
116
+ # Обрати внимание: теперь функция возвращает 2 значения: логи и медиа (поэтому outputs=[logs_output, ...])
117
+ btn_photo.click(
118
+ fn=create_photo,
119
+ inputs=[blogger_imgs, product_imgs],
120
+ outputs=[logs_output, out_photo]
121
+ )
122
+
123
+ btn_video.click(
124
+ fn=create_video,
125
+ inputs=[blogger_imgs, product_imgs],
126
+ outputs=[logs_output, out_video]
127
+ )
128
+
129
+ if __name__ == "__main__":
130
+ demo.launch()
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ gradio>=4.0.0
2
+ requests
3
+ fal-client