Update index.html

index.html

@@ -1,15 +1,15 @@
 <!DOCTYPE html>
-<html lang="en" translate="no">
+<html lang="ru" translate="no">
 <head>
-  <meta charset="utf-8">
+  <meta charset="UTF-8">
   <meta name="google" content="notranslate">
 
-  <script type="module" src="https://cdn.jsdelivr.net/npm/@gradio/lite/dist/lite.js"></script>
-
-  <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@gradio/lite/dist/lite.css" />
+  <script type="module"
+    src="https://cdn.jsdelivr.net/npm/@gradio/lite/dist/lite.js">
+  </script>
 </head>
 
-<body style="background-color: #0f172a; color: white; padding: 15px;">
+<body>
 
 <gradio-app>
 
@@ -20,139 +20,142 @@ import json
 from js import fetch
 
 
-async def generate_img(prompt):
+async def generate_image(prompt):
 
-    if not prompt:
-        return None, "Введите описание идеи!"
+    if not prompt or not prompt.strip():
+        return None, "❌ Напиши, что нужно создать"
 
     try:
-
         response = await fetch(
             "https://mrfakename-z-image-turbo.hf.space/call/predict",
-            method="POST",
-            headers={
-                "Content-Type": "application/json"
-            },
-            body=json.dumps({
-                "data": [prompt]
-            })
+            {
+                "method": "POST",
+                "headers": {
+                    "Content-Type": "application/json"
+                },
+                "body": json.dumps({
+                    "data": [prompt]
+                })
+            }
         )
 
-        res_data = await response.json()
+        data = await response.json()
+        event_id = data.get("event_id")
 
-        event_id = res_data.get("event_id")
+        if not event_id:
+            return None, "❌ Сервер не вернул event_id"
 
-        res_event = await fetch(
+        result = await fetch(
             f"https://mrfakename-z-image-turbo.hf.space/call/predict/{event_id}"
         )
 
-        text_data = await res_event.text()
+        text = await result.text()
 
-        for line in text_data.split("\n"):
+        for line in text.splitlines():
 
             if line.startswith("data:"):
 
-                img_json = json.loads(line[5:])
+                raw = line[5:].strip()
 
-                return (
-                    img_json[0][0]["url"],
-                    "Успешно сгенерировано!"
-                )
+                if not raw:
+                    continue
 
-        return None, "Ошибка получения данных."
+                result_data = json.loads(raw)
 
-    except Exception as e:
+                try:
+                    image_url = result_data[0][0]["url"]
+                    return image_url, "✅ Изображение готово!"
+                except:
+                    pass
+
+        return None, "❌ Не удалось получить изображение"
 
-        return None, f"Ошибка: {str(e)}"
+    except Exception as e:
+        return None, "❌ Ошибка: " + str(e)
 
 
-async def auto_run(idea):
+def improve_prompt(idea):
 
-    if not idea:
-        return "", None, "Введите идею!"
+    if not idea or not idea.strip():
+        return ""
 
-    prompt = (
-        f"{idea}, "
-        "cinematic composition, "
-        "photorealistic, "
-        "highly detailed, "
-        "realistic lighting, "
-        "realistic shadows, "
-        "natural colors, "
-        "detailed environment, "
-        "professional photography, "
-        "sharp focus, "
-        "depth of field, "
-        "8K quality"
+    return (
+        idea.strip()
+        + ", cinematic composition, "
+        + "highly detailed, photorealistic, "
+        + "realistic lighting, realistic shadows, "
+        + "detailed environment, sharp focus, "
+        + "depth of field, professional photography, "
+        + "high quality"
     )
 
-    img, status = await generate_img(prompt)
 
-    return prompt, img, f"[АВТО] {status}"
+async def auto_generate(idea):
 
+    if not idea or not idea.strip():
+        return "", None, "❌ Напиши идею"
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    prompt = improve_prompt(idea)
 
-    gr.Markdown("# 🤖 MODARA_Al")
+    image, status = await generate_image(prompt)
 
-    gr.Markdown("Ваш личный AI-агент")
+    return prompt, image, status
 
-    with gr.Row():
 
-        user_idea = gr.Textbox(
-            label="Что создать?",
-            placeholder="BMW M5 CS едет по ночному Дубаю"
-        )
+with gr.Blocks() as app:
+
+    gr.Markdown(
+        """
+        # 🤖 MODARA_Al
+        ### Твой личный AI-агент
+        """
+    )
+
+    idea = gr.Textbox(
+        label="💡 Что создать?",
+        placeholder="Например: BMW M5 CS едет ночью по Дубаю",
+        lines=3
+    )
 
     with gr.Row():
 
-        btn_auto = gr.Button(
+        auto_button = gr.Button(
             "⚡ АВТО",
             variant="primary"
         )
 
-        btn_img = gr.Button(
-            "🎨 Только картинка"
+        image_button = gr.Button(
+            "🎨 СОЗДАТЬ"
         )
 
-    status_box = gr.Textbox(
-        label="Статус",
+    status = gr.Textbox(
+        label="📊 Статус",
         interactive=False
     )
 
-    final_prompt = gr.Textbox(
-        label="Улучшенный промпт",
-        interactive=False
+    improved = gr.Textbox(
+        label="✨ Улучшенный промпт",
+        interactive=False,
+        lines=5
     )
 
-    output_image = gr.Image(
-        label="Результат"
+    result = gr.Image(
+        label="🖼️ Результат"
     )
 
 
-    btn_img.click(
-        fn=generate_img,
-        inputs=[user_idea],
-        outputs=[
-            output_image,
-            status_box
-        ]
+    auto_button.click(
+        auto_generate,
+        inputs=idea,
+        outputs=[improved, result, status]
     )
 
-
-    btn_auto.click(
-        fn=auto_run,
-        inputs=[user_idea],
-        outputs=[
-            final_prompt,
-            output_image,
-            status_box
-        ]
+    image_button.click(
+        generate_image,
+        inputs=idea,
+        outputs=[result, status]
     )
 
-
-demo.launch()
-
 </script>
 
 </gradio-app>