added seed during generate

app.py

@@ -88,13 +88,16 @@ def read_root():
 
 
 @app.get("/generate")
-def generate_random():
-    """Endpoint 1: Purely random generation for 'Discovery'."""
+def generate_random(seed: int = Query(None)):
+    """Endpoint 1: Fixed context generation for a session."""
     if gen_model is None: raise HTTPException(status_code=503)
     
-    with torch.inference_mode():        
+    # Use the provided session seed or fallback to random
+    active_seed = seed if seed is not None else torch.seed()
+    torch.manual_seed(active_seed)
+    
+    with torch.inference_mode():
         noise = torch.randn(1, Z_DIM, device=DEVICE)
-        print("NOISE:", noise[0, :5])
         fake_img = gen_model(noise)
         return StreamingResponse(get_image_stream(fake_img), media_type="image/png")