One last time 001

services/ai_service.py

@@ -105,9 +105,20 @@ class AIService:
                 import shutil
                 shutil.copy(generated_path, destination)
                 
-                # Convert absolute path to local URL for frontend
-                # assuming running on localhost:8000
-                return f"http://localhost:8000/static/{output_filename}"
+                # Dynamic URL Generation for Deployment
+                # If running on Hugging Face, SPACE_ID (e.g. "user/repo") is set.
+                space_id = os.getenv("SPACE_ID")
+                if space_id:
+                    # HF Spaces use the format: https://user-repo.hf.space
+                    host = space_id.replace("/", "-")
+                    base_url = f"https://{host}.hf.space"
+                else:
+                    # Localhost fallback
+                    base_url = "http://localhost:8000"
+                
+                final_url = f"{base_url}/static/{output_filename}"
+                print(f"✅ Generated Image URL: {final_url}")
+                return final_url
                 
             return None