Update app.py

app.py

@@ -1,7 +1,11 @@
 import os
 import gradio as gr
+import torch
 from openai import OpenAI
 
+from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
+
+# ----------------- CHAT (HF Router) -----------------
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 client = OpenAI(
@@ -9,12 +13,12 @@ client = OpenAI(
     api_key=HF_TOKEN,
 )
 
-MODEL_ID = "HuggingFaceTB/SmolLM3-3B:hf-inference"
+CHAT_MODEL = "HuggingFaceTB/SmolLM3-3B:hf-inference"
 SYSTEM = "Sen MAIND AI'sin. Kısa, net ve yardımcı cevap ver."
 
 def llm_reply(user_text: str) -> str:
     resp = client.chat.completions.create(
-        model=MODEL_ID,
+        model=CHAT_MODEL,
         messages=[
             {"role": "system", "content": SYSTEM},
             {"role": "user", "content": user_text},
@@ -30,111 +34,77 @@ def chat_fn(message, history):
     try:
         reply = llm_reply(message)
     except Exception as e:
-        reply = f"⚠️ Hata: {e}"
+        reply = f"⚠️ Chat hata: {e}"
     history.append({"role": "assistant", "content": reply})
     return history, ""
 
-CSS = """
-:root{
-  --bg:#050b1a;
-  --panel:#0a1636;
-  --panel2:#0c1d46;
-  --stroke:rgba(56,189,248,.25);
-  --glow:rgba(56,189,248,.25);
-  --txt:#e6f1ff;
-  --muted:#93c5fd;
-  --blue:#38bdf8;
-}
-body, .gradio-container{
-  background: radial-gradient(1200px 800px at 20% 0%, rgba(56,189,248,.18), transparent 55%),
-              radial-gradient(900px 700px at 90% 20%, rgba(99,102,241,.18), transparent 55%),
-              var(--bg) !important;
-  color: var(--txt) !important;
-}
-#wrap{
-  max-width: 980px;
-  margin: 0 auto;
-}
-#topbar{
-  display:flex; align-items:center; gap:14px;
-  padding: 14px 16px;
-  background: linear-gradient(180deg, rgba(10,22,54,.9), rgba(10,22,54,.55));
-  border: 1px solid var(--stroke);
-  border-radius: 16px;
-  box-shadow: 0 0 0 1px rgba(56,189,248,.06), 0 20px 60px rgba(0,0,0,.35);
-}
-#brand h1{ margin:0; font-size: 20px; letter-spacing:.4px; }
-#brand p{ margin:2px 0 0; color: var(--muted); font-size: 12px; }
+# ----------------- IMAGE (LOCAL in Space, CPU) -----------------
+IMG_MODEL = "segmind/tiny-sd"  # küçük ve Space için uygun :contentReference[oaicite:1]{index=1}
+img_pipe = None
 
-#card{
-  margin-top: 14px;
-  padding: 14px;
-  background: linear-gradient(180deg, rgba(12,29,70,.85), rgba(10,22,54,.6));
-  border: 1px solid var(--stroke);
-  border-radius: 18px;
-  box-shadow: 0 0 0 1px rgba(56,189,248,.05), 0 30px 90px rgba(0,0,0,.45);
-}
+def get_pipe():
+    global img_pipe
+    if img_pipe is None:
+        pipe = StableDiffusionPipeline.from_pretrained(
+            IMG_MODEL,
+            torch_dtype=torch.float32,
+        )
+        pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+        pipe.enable_attention_slicing()   # RAM azaltır
+        pipe = pipe.to("cpu")
+        img_pipe = pipe
+    return img_pipe
 
-/* Chat area */
-#chat .wrap{
-  border-radius: 14px;
-}
-.gr-chatbot{
-  background: transparent !important;
-}
-.gr-chatbot .message{
-  border-radius: 14px !important;
-  border: 1px solid rgba(56,189,248,.18) !important;
-  box-shadow: 0 0 20px rgba(56,189,248,.08);
-}
-.gr-chatbot .message.user{
-  background: rgba(56,189,248,.08) !important;
-}
-.gr-chatbot .message.bot{
-  background: rgba(99,102,241,.08) !important;
-}
+def gen_image(prompt: str, steps: int, size: int):
+    if not prompt or not prompt.strip():
+        raise gr.Error("Prompt boş.")
+    pipe = get_pipe()
+    with torch.inference_mode():
+        out = pipe(
+            prompt=prompt,
+            num_inference_steps=int(steps),
+            guidance_scale=6.5,
+            height=int(size),
+            width=int(size),
+        )
+    return out.images[0]
 
-/* Input row */
-#inputrow{
-  margin-top: 10px;
-  gap: 10px;
-}
-#sendbtn button{
-  background: linear-gradient(90deg, rgba(56,189,248,.95), rgba(99,102,241,.95)) !important;
-  border: 1px solid rgba(56,189,248,.35) !important;
-  border-radius: 14px !important;
-  box-shadow: 0 0 30px rgba(56,189,248,.18);
-}
-#sendbtn button:hover{
-  filter: brightness(1.05);
-}
-textarea, input, .gr-text-input textarea{
-  background: rgba(5,11,26,.55) !important;
-  border: 1px solid rgba(56,189,248,.22) !important;
-  border-radius: 14px !important;
-  color: var(--txt) !important;
-}
+# ----------------- UI (mavi neon) -----------------
+CSS = """
+:root{--bg:#050b1a;--panel:#0a1636;--stroke:rgba(56,189,248,.25);--txt:#e6f1ff;--muted:#93c5fd;}
+body,.gradio-container{background:radial-gradient(1200px 800px at 20% 0%, rgba(56,189,248,.18), transparent 55%),var(--bg)!important;color:var(--txt)!important;}
+#wrap{max-width:1100px;margin:0 auto;}
+#topbar{display:flex;align-items:center;gap:14px;padding:14px 16px;background:rgba(10,22,54,.75);border:1px solid var(--stroke);border-radius:16px;}
+#card{margin-top:14px;padding:14px;background:rgba(10,22,54,.55);border:1px solid var(--stroke);border-radius:18px;}
+button{border-radius:14px!important;}
+textarea,input{background:rgba(5,11,26,.55)!important;border:1px solid rgba(56,189,248,.22)!important;color:var(--txt)!important;}
 """
 
 with gr.Blocks(css=CSS, title="MaindAI") as demo:
     with gr.Column(elem_id="wrap"):
         with gr.Row(elem_id="topbar"):
             gr.Image("logo.png", show_label=False, height=58, width=58, container=False)
-            with gr.Column(elem_id="brand"):
-                gr.Markdown("# MAIND AI")
-                gr.Markdown("<p>Neon mavi tema • HF Router • SmolLM3-3B</p>")
+            gr.Markdown("## 💙 MAIND AI\n<small style='color:#93c5fd'>Chat + Görsel (HF Space CPU)</small>")
 
-        with gr.Column(elem_id="card"):
-            chatbot = gr.Chatbot(value=[], elem_id="chat", height=420)
-            with gr.Row(elem_id="inputrow"):
-                msg = gr.Textbox(
-                    placeholder="Bir şey sor…",
-                    show_label=False,
-                    scale=8
-                )
-                send = gr.Button("Gönder", elem_id="sendbtn", scale=2)
+        with gr.Row(elem_id="card"):
+            # SOL: Chat
+            with gr.Column(scale=6):
+                gr.Markdown("### 💬 Chat")
+                chatbot = gr.Chatbot(value=[], height=420)
+                msg = gr.Textbox(placeholder="Bir şey sor…", show_label=False)
+                send = gr.Button("Gönder")
+                send.click(chat_fn, [msg, chatbot], [chatbot, msg])
+                msg.submit(chat_fn, [msg, chatbot], [chatbot, msg])
 
-            send.click(chat_fn, [msg, chatbot], [chatbot, msg])
-            msg.submit(chat_fn, [msg, chatbot], [chatbot, msg])
+            # SAĞ: Image
+            with gr.Column(scale=5):
+                gr.Markdown("### 🎨 Görsel Üret")
+                img_prompt = gr.Textbox(placeholder="Örn: neon mavi cyberpunk şehir, yağmur, gece", show_label=False)
+                with gr.Row():
+                    steps = gr.Slider(2, 12, value=6, step=1, label="Steps (hız)")
+                    size = gr.Slider(256, 512, value=384, step=64, label="Boyut")
+                img_btn = gr.Button("Görsel Oluştur")
+                img_out = gr.Image(height=420)
+                img_btn.click(gen_image, [img_prompt, steps, size], img_out)
 
 demo.launch()