Update app.py

app.py

@@ -1,22 +1,83 @@
 import gradio as gr
+from PIL import Image, ImageDraw
+
+
+def fake_generate(prompt, negative_prompt, width, height, steps, guidance_scale):
+    # Временная заглушка, чтобы тренироваться с интерфейсом Space
+    img = Image.new("RGB", (width, height), color=(30, 30, 30))
+    draw = ImageDraw.Draw(img)
+
+    text = (
+        "DepthPaint Studio\n\n"
+        "TXT2IMG prototype is alive.\n\n"
+        f"Prompt:\n{prompt[:120]}\n\n"
+        f"Negative:\n{negative_prompt[:120]}\n\n"
+        f"Steps: {steps}\n"
+        f"CFG: {guidance_scale}"
+    )
+    draw.multiline_text((20, 20), text, fill=(220, 220, 220), spacing=6)
+
+    return img
+
 
 with gr.Blocks() as demo:
     gr.Markdown("# DepthPaint Studio")
-    gr.Markdown("Prototype is alive. Stable Diffusion inpaint will be connected next.")
+    gr.Markdown("Simple Stable Diffusion prompt generation prototype.")
+
+    with gr.Row():
+        with gr.Column():
+            prompt = gr.Textbox(
+                label="Prompt",
+                lines=4,
+                value="A cozy fantasy workshop, warm lighting, detailed illustration"
+            )
+
+            negative_prompt = gr.Textbox(
+                label="Negative prompt",
+                lines=3,
+                value="blurry, low quality, distorted, ugly"
+            )
+
+            with gr.Row():
+                width = gr.Slider(
+                    minimum=256,
+                    maximum=1024,
+                    step=64,
+                    value=512,
+                    label="Width"
+                )
+                height = gr.Slider(
+                    minimum=256,
+                    maximum=1024,
+                    step=64,
+                    value=512,
+                    label="Height"
+                )
 
-    image = gr.Image(type="pil", label="Input image")
-    mask = gr.Image(type="pil", label="Mask image")
-    prompt = gr.Textbox(label="Prompt", value="a clean restored image")
+            with gr.Row():
+                steps = gr.Slider(
+                    minimum=1,
+                    maximum=50,
+                    step=1,
+                    value=20,
+                    label="Steps"
+                )
+                guidance_scale = gr.Slider(
+                    minimum=1.0,
+                    maximum=15.0,
+                    step=0.5,
+                    value=7.5,
+                    label="Guidance scale"
+                )
 
-    run_btn = gr.Button("Generate")
-    output = gr.Image(label="Output")
+            generate_btn = gr.Button("Generate")
 
-    def fake_inpaint(img, msk, text):
-        return img
+        with gr.Column():
+            output = gr.Image(label="Generated image")
 
-    run_btn.click(
-        fn=fake_inpaint,
-        inputs=[image, mask, prompt],
+    generate_btn.click(
+        fn=fake_generate,
+        inputs=[prompt, negative_prompt, width, height, steps, guidance_scale],
         outputs=output,
     )