Update index.html

index.html

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