Spaces:

Valtry
/

My-Image

Sleeping

App Files Files Community

Valtry commited on Aug 8, 2025

Commit

7c9d195

verified ·

1 Parent(s): 67c9c6c

Update app.py

Browse files

Files changed (1) hide show

app.py +40 -32

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
-# Set cache directories to writable paths
 os.environ["HF_HOME"] = "/app/cache"
 os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
 os.makedirs("/app/cache", exist_ok=True)
@@ -20,12 +20,12 @@ from diffusers import StableDiffusionPipeline
 MODEL_ID = "runwayml/stable-diffusion-v1-5"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 STATIC_FOLDER = "/app/static"
-SPACE_URL = "https://valtry-my-image.hf.space"  # <-- CHANGE THIS to your Space's URL
 # ------------------------
 app = FastAPI(title="Valtry Text→Image API")
-# Serve static folder
 app.mount("/static", StaticFiles(directory=STATIC_FOLDER), name="static")
 print(f"Loading model {MODEL_ID} on {DEVICE}...")
@@ -45,7 +45,7 @@ class GenerateReq(BaseModel):
 @app.post("/generate")
 async def generate(req: GenerateReq):
-    if not req.prompt.strip():
         return JSONResponse({"error": "prompt is required"}, status_code=400)
     seed = req.seed if req.seed is not None else int(time.time() * 1000) % 2**32
@@ -67,68 +67,76 @@ async def generate(req: GenerateReq):
     file_path = os.path.join(STATIC_FOLDER, filename)
     image.save(file_path)
-    # ✅ Full public URL so it loads in browser
     public_url = f"{SPACE_URL}/static/{filename}"
     return {"url": public_url, "filename": filename}
 @app.get("/", response_class=HTMLResponse)
 async def home():
-    return f"""
     <!doctype html>
     <html>
     <head>
       <meta charset="utf-8"/>
-      <title>Valtry — Text→Image</title>
       <style>
-        body{{font-family:Arial,sans-serif;margin:32px;background:#f7f7f7}}
-        input, button, textarea{{font-size:16px;padding:10px;width:100%;margin-top:8px}}
-        img{{max-width:100%;border:1px solid #ccc;padding:6px;background:#fff;margin-top:20px}}
       </style>
     </head>
     <body>
       <h2>Valtry — Text → Image</h2>
       <textarea id="prompt" rows="3" placeholder="A fantasy castle on a cliff at sunset"></textarea><br>
-      <input id="steps" type="number" value="25" min="1" max="150"/>
-      <input id="scale" type="number" value="7.5" step="0.1" min="1" max="20"/>
-      <input id="seed" type="number" placeholder="optional seed"/>
       <button onclick="generate()">Generate Image</button>
       <div id="status"></div>
       <div id="result"></div>
       <script>
-        async function generate(){{
           const prompt = document.getElementById('prompt').value;
-          const steps = parseInt(document.getElementById('steps').value);
-          const scale = parseFloat(document.getElementById('scale').value);
           const seedVal = document.getElementById('seed').value;
-          document.getElementById('status').textContent = "⏳ Generating...";
           document.getElementById('result').innerHTML = "";
-          const body = {{ prompt, num_inference_steps: steps, guidance_scale: scale }};
           if (seedVal) body.seed = parseInt(seedVal);
-          try {{
-            const res = await fetch('/generate', {{
               method: 'POST',
-              headers: {{ 'Content-Type': 'application/json' }},
               body: JSON.stringify(body)
-            }});
             const data = await res.json();
-            if (res.ok) {{
-              document.getElementById('status').textContent = "✅ Done";
-              document.getElementById('result').innerHTML = `<img src="${data.url}" alt="Generated image"/>`;
-            }} else {{
-              document.getElementById('status').textContent = "❌ Error: " + data.error;
-            }}
-          }} catch (err) {{
-            document.getElementById('status').textContent = "❌ " + err.message;
-          }}
-        }}
       </script>
     </body>
     </html>
     """
 @app.get("/health")
 async def health():

 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
+# Make caches and static directories writable
 os.environ["HF_HOME"] = "/app/cache"
 os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
 os.makedirs("/app/cache", exist_ok=True)
 MODEL_ID = "runwayml/stable-diffusion-v1-5"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 STATIC_FOLDER = "/app/static"
+SPACE_URL = "https://valtry-my-image.hf.space"   # <- set your space URL here
 # ------------------------
 app = FastAPI(title="Valtry Text→Image API")
+# Serve static files publicly at /static/...
 app.mount("/static", StaticFiles(directory=STATIC_FOLDER), name="static")
 print(f"Loading model {MODEL_ID} on {DEVICE}...")
 @app.post("/generate")
 async def generate(req: GenerateReq):
+    if not req.prompt or not req.prompt.strip():
         return JSONResponse({"error": "prompt is required"}, status_code=400)
     seed = req.seed if req.seed is not None else int(time.time() * 1000) % 2**32
     file_path = os.path.join(STATIC_FOLDER, filename)
     image.save(file_path)
+    # Return an absolute public URL (so external pages can load it)
     public_url = f"{SPACE_URL}/static/{filename}"
     return {"url": public_url, "filename": filename}
+# Home page: NOTE -> regular string (NOT an f-string) to avoid Python interpolating JS {..}
 @app.get("/", response_class=HTMLResponse)
 async def home():
+    html = """
     <!doctype html>
     <html>
     <head>
       <meta charset="utf-8"/>
+      <title>Valtry — Text → Image</title>
       <style>
+        body{font-family:Arial,sans-serif;margin:32px;background:#f7f7f7}
+        textarea,input,button{font-size:16px;padding:10px;width:100%;margin-top:8px;box-sizing:border-box}
+        img{max-width:100%;border:1px solid #ccc;padding:6px;background:#fff;margin-top:20px}
       </style>
     </head>
     <body>
       <h2>Valtry — Text → Image</h2>
       <textarea id="prompt" rows="3" placeholder="A fantasy castle on a cliff at sunset"></textarea><br>
+      <label>Steps (num_inference_steps)</label>
+      <input id="steps" type="number" value="25" min="1" max="150"/><br>
+      <label>Guidance scale</label>
+      <input id="scale" type="number" value="7.5" step="0.1" min="1" max="20"/><br>
+      <label>Seed (optional)</label>
+      <input id="seed" type="number" placeholder="optional seed"/><br>
       <button onclick="generate()">Generate Image</button>
       <div id="status"></div>
       <div id="result"></div>
       <script>
+        async function generate(){
           const prompt = document.getElementById('prompt').value;
+          const steps = parseInt(document.getElementById('steps').value || 25);
+          const scale = parseFloat(document.getElementById('scale').value || 7.5);
           const seedVal = document.getElementById('seed').value;
+          document.getElementById('status').textContent = "⏳ Generating — this may take a bit...";
           document.getElementById('result').innerHTML = "";
+          const body = { prompt: prompt, num_inference_steps: steps, guidance_scale: scale };
           if (seedVal) body.seed = parseInt(seedVal);
+          try {
+            const res = await fetch('/generate', {
               method: 'POST',
+              headers: { 'Content-Type': 'application/json' },
               body: JSON.stringify(body)
+            });
+            if (!res.ok) {
+              const txt = await res.text();
+              document.getElementById('status').textContent = '❌ Error ' + res.status + ': ' + txt;
+              return;
+            }
             const data = await res.json();
+            document.getElementById('status').textContent = '✅ Done — image below';
+            document.getElementById('result').innerHTML = `<img src="${data.url}" alt="generated-image"/>`;
+          } catch (err) {
+            document.getElementById('status').textContent = '❌ Exception: ' + err.message;
+          }
+        }
       </script>
     </body>
     </html>
     """
+    return HTMLResponse(content=html)
 @app.get("/health")
 async def health():