zaebalo blyat

app.py

@@ -307,7 +307,49 @@ def format_wd14_tags(tags: Dict[str, float], threshold: float = 0.35) -> str:
 def generate_3d(scribble_image_dict, prompt, scribble_confidence, text_confidence, seed):
     print("Generating 3D model...")
     input_prompt = prompt # Keep track of original prompt for return on early exit
-    if scribble_image_dict is None or scribble_image_dict.get("composite") is None:
+    if scribble_image_dict is None:
+        print("No scribble image provided.")
+        return None, input_prompt # Return None for model, original prompt
+
+    # ImageEditor can return composite=None for API calls; rebuild from background/layers if needed.
+    def _image_from_value(value):
+        if value is None:
+            return None
+        if isinstance(value, Image.Image):
+            return value
+        if isinstance(value, np.ndarray):
+            return Image.fromarray(value)
+        if isinstance(value, str):
+            try:
+                return Image.open(value)
+            except Exception:
+                return None
+        return None
+
+    def _build_composite(image_dict):
+        composite_img = _image_from_value(image_dict.get("composite"))
+        if composite_img is not None:
+            return composite_img
+
+        background = _image_from_value(image_dict.get("background"))
+        layers = image_dict.get("layers") or []
+        layer_images = [_image_from_value(layer) for layer in layers]
+        layer_images = [img for img in layer_images if img is not None]
+
+        if background is None and not layer_images:
+            return None
+
+        if background is None:
+            base = Image.new("RGBA", layer_images[0].size, (255, 255, 255, 255))
+        else:
+            base = background.convert("RGBA")
+
+        for layer_img in layer_images:
+            base = Image.alpha_composite(base, layer_img.convert("RGBA"))
+        return base.convert("RGB")
+
+    composite_image = _build_composite(scribble_image_dict)
+    if composite_image is None:
         print("No scribble image provided.")
         return None, input_prompt # Return None for model, original prompt
 
@@ -317,7 +359,7 @@ def generate_3d(scribble_image_dict, prompt, scribble_confidence, text_confidenc
         print("Prompt is empty, attempting WD14 tagging...")
         try:
             # Get the user drawing (black on white) for tagging
-            user_drawing_img = Image.fromarray(scribble_image_dict["composite"]).convert("RGB")
+            user_drawing_img = composite_image.convert("RGB")
             tag_results = wd14_tagger.interrogate(user_drawing_img)
             if tag_results:
                 ratings, tags = tag_results
@@ -346,7 +388,7 @@ def generate_3d(scribble_image_dict, prompt, scribble_confidence, text_confidenc
     # --- Image Preprocessing for TripoSG ---
     # Get the composite image again (safer in case dict is modified)
     # The composite might be RGBA if a layer was involved, ensure RGB for processing
-    image_for_triposg = Image.fromarray(scribble_image_dict["composite"]).convert("RGB")
+    image_for_triposg = composite_image.convert("RGB")
     # Preprocess the image: invert colors (black on white -> white on black)
     image_np = np.array(image_for_triposg)
     processed_image_np = 255 - image_np

local_service.py

@@ -0,0 +1,123 @@
+from __future__ import annotations
+
+import os
+import tempfile
+import uuid
+from typing import Dict
+
+from fastapi import FastAPI, File, Form, UploadFile
+from fastapi.responses import HTMLResponse, FileResponse, PlainTextResponse, RedirectResponse
+from fastapi.staticfiles import StaticFiles
+from gradio_client import Client, handle_file
+from PIL import Image
+from jinja2 import Environment, FileSystemLoader, select_autoescape
+
+APP_TITLE = "Text Sobirai - Local Viewer"
+HF_SPACE = "kendaaa/text_sobirai"
+
+app = FastAPI(title=APP_TITLE)
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+jinja_env = Environment(
+    loader=FileSystemLoader("templates"),
+    autoescape=select_autoescape(["html", "xml"]),
+)
+
+hf_token = os.environ.get("HF_TOKEN")
+client = Client(HF_SPACE, hf_token=hf_token) if hf_token else Client(HF_SPACE)
+
+model_store: Dict[str, str] = {}
+
+
+def render_template(name: str, **context: object) -> HTMLResponse:
+    template = jinja_env.get_template(name)
+    return HTMLResponse(template.render(**context))
+
+
+@app.get("/", response_class=HTMLResponse)
+def index() -> HTMLResponse:
+    return render_template("index.html", title=APP_TITLE)
+
+
+@app.post("/generate", response_class=HTMLResponse)
+def generate(
+    prompt: str = Form(...),
+    scribble: UploadFile | None = File(None),
+    scribble_confidence: float = Form(0.4),
+    text_confidence: float = Form(1.0),
+    seed: int = Form(0),
+) -> HTMLResponse:
+    if scribble is None or not scribble.filename:
+        background = Image.new("RGB", (256, 256), "white")
+        layer = Image.new("RGBA", background.size, (0, 0, 0, 0))
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as scribble_tmp:
+            scribble_path = scribble_tmp.name
+            layer.save(scribble_path)
+    else:
+        suffix = os.path.splitext(scribble.filename or "scribble.png")[1] or ".png"
+        with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as scribble_tmp:
+            scribble_path = scribble_tmp.name
+            scribble_tmp.write(scribble.file.read())
+        try:
+            with Image.open(scribble_path) as img:
+                background = Image.new("RGB", img.size, "white")
+        except Exception:
+            background = Image.new("RGB", (256, 256), "white")
+
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as bg_tmp:
+        background_path = bg_tmp.name
+        background.save(background_path)
+
+    result = client.predict(
+        scribble_image_dict={
+            "background": handle_file(background_path),
+            "layers": [handle_file(scribble_path)],
+            "composite": None,
+            "id": None,
+        },
+        prompt=prompt,
+        scribble_confidence=float(scribble_confidence),
+        text_confidence=float(text_confidence),
+        seed=int(seed),
+        api_name="/generate_3d",
+    )
+
+    model_path, used_prompt = result
+    if not model_path:
+        return render_template(
+            "result.html",
+            title=APP_TITLE,
+            error="Model was not generated. Check the Space logs.",
+            prompt=used_prompt or prompt,
+            model_url=None,
+        )
+
+    model_id = str(uuid.uuid4())
+    model_store[model_id] = model_path
+    model_url = f"/models/{model_id}"
+    return render_template(
+        "result.html",
+        title=APP_TITLE,
+        error=None,
+        prompt=used_prompt or prompt,
+        model_url=model_url,
+    )
+
+
+@app.get("/models/{model_id}")
+def get_model(model_id: str):
+    path = model_store.get(model_id)
+    if not path:
+        return PlainTextResponse("model not found", status_code=404)
+    return FileResponse(path, media_type="model/gltf-binary")
+
+
+@app.get("/generate")
+def generate_get() -> RedirectResponse:
+    return RedirectResponse(url="/")
+
+
+if __name__ == "__main__":
+    import uvicorn
+
+    uvicorn.run("local_service:app", host="0.0.0.0", port=8000, reload=True)

requirements.local.txt

@@ -0,0 +1,6 @@
+fastapi
+uvicorn
+jinja2
+python-multipart
+gradio_client
+pillow

static/styles.css

@@ -0,0 +1,100 @@
+:root {
+  --bg: #f7f4f0;
+  --panel: #ffffff;
+  --text: #1d1b1a;
+  --accent: #2b6f65;
+  --border: #e5e0d8;
+}
+
+* {
+  box-sizing: border-box;
+}
+
+body {
+  margin: 0;
+  font-family: "Iowan Old Style", "Garamond", "Times New Roman", serif;
+  color: var(--text);
+  background: linear-gradient(135deg, #f7f4f0, #efe9e1);
+}
+
+.page {
+  max-width: 980px;
+  margin: 0 auto;
+  padding: 40px 20px 60px;
+}
+
+.hero {
+  text-align: center;
+  margin-bottom: 32px;
+}
+
+.hero h1 {
+  font-size: 2.4rem;
+  margin: 0 0 8px;
+}
+
+.panel {
+  background: var(--panel);
+  border: 1px solid var(--border);
+  border-radius: 16px;
+  padding: 24px;
+  box-shadow: 0 20px 40px rgba(0, 0, 0, 0.06);
+}
+
+.form {
+  display: grid;
+  gap: 16px;
+}
+
+.field {
+  display: grid;
+  gap: 6px;
+  font-size: 0.95rem;
+}
+
+.row {
+  display: grid;
+  gap: 16px;
+  grid-template-columns: repeat(auto-fit, minmax(180px, 1fr));
+}
+
+input[type="text"],
+input[type="number"],
+input[type="file"] {
+  padding: 10px 12px;
+  border-radius: 10px;
+  border: 1px solid var(--border);
+  font-size: 1rem;
+}
+
+button {
+  background: var(--accent);
+  color: white;
+  border: none;
+  padding: 12px 18px;
+  border-radius: 12px;
+  font-size: 1rem;
+  cursor: pointer;
+}
+
+button:hover {
+  opacity: 0.92;
+}
+
+.viewer {
+  width: 100%;
+  height: 520px;
+  background: #fafafa;
+  border-radius: 12px;
+}
+
+.error {
+  color: #9a2c2c;
+  font-weight: 600;
+}
+
+@media (max-width: 640px) {
+  .viewer {
+    height: 360px;
+  }
+}

templates/index.html

@@ -0,0 +1,45 @@
+<!doctype html>
+<html lang="ru">
+<head>
+  <meta charset="utf-8" />
+  <meta name="viewport" content="width=device-width, initial-scale=1" />
+  <title>{{ title }}</title>
+  <link rel="stylesheet" href="/static/styles.css" />
+</head>
+<body>
+  <main class="page">
+    <header class="hero">
+      <h1>{{ title }}</h1>
+      <p>Локальный просмотр 3D модели из Space на Hugging Face.</p>
+    </header>
+
+    <section class="panel">
+      <form class="form" action="/generate" method="post" enctype="multipart/form-data">
+        <label class="field">
+          <span>Prompt</span>
+          <input type="text" name="prompt" placeholder="Например: a small chair" required />
+        </label>
+        <label class="field">
+          <span>Scribble (PNG, опционально)</span>
+          <input type="file" name="scribble" accept="image/*" />
+        </label>
+        <div class="row">
+          <label class="field">
+            <span>Scribble confidence</span>
+            <input type="number" name="scribble_confidence" step="0.05" min="0" max="1" value="0.4" />
+          </label>
+          <label class="field">
+            <span>Prompt confidence</span>
+            <input type="number" name="text_confidence" step="0.05" min="0" max="1" value="1.0" />
+          </label>
+          <label class="field">
+            <span>Seed</span>
+            <input type="number" name="seed" value="0" />
+          </label>
+        </div>
+        <button type="submit">Сгенерировать</button>
+      </form>
+    </section>
+  </main>
+</body>
+</html>

templates/result.html

@@ -0,0 +1,34 @@
+<!doctype html>
+<html lang="ru">
+<head>
+  <meta charset="utf-8" />
+  <meta name="viewport" content="width=device-width, initial-scale=1" />
+  <title>{{ title }}</title>
+  <link rel="stylesheet" href="/static/styles.css" />
+  <script type="module" src="https://unpkg.com/@google/model-viewer/dist/model-viewer.min.js"></script>
+</head>
+<body>
+  <main class="page">
+    <header class="hero">
+      <h1>{{ title }}</h1>
+      <p>Prompt: <strong>{{ prompt }}</strong></p>
+      <p><a href="/">Сгенерировать еще</a></p>
+    </header>
+
+    <section class="panel">
+      {% if error %}
+        <p class="error">{{ error }}</p>
+      {% else %}
+        <model-viewer
+          class="viewer"
+          src="{{ model_url }}"
+          auto-rotate
+          camera-controls
+          shadow-intensity="0.8"
+          exposure="1.0"
+        ></model-viewer>
+      {% endif %}
+    </section>
+  </main>
+</body>
+</html>