Update main.py

main.py

@@ -407,13 +407,48 @@ async def playground():
     if html_content is None:
         return HTMLResponse(content="<h1>image-playground.html not found</h1>", status_code=404)
     return HTMLResponse(content=html_content)
+
+
+
+
+# VETRA
+GITHUB_BASE = "https://raw.githubusercontent.com/Parthsadaria/Vetra/main"
+
+FILES = {
+    "html": "index.html",
+    "css": "style.css",
+    "js": "script.js"
+}
+
+async def get_github_file(filename: str) -> str:
+    url = f"{GITHUB_BASE}/{filename}"
+    async with httpx.AsyncClient() as client:
+        res = await client.get(url)
+        return res.text if res.status_code == 200 else None
+
 @app.get("/vetra", response_class=HTMLResponse)
-async def playground():
-    html_content = read_html_file("Vetra.html")
-    if html_content is None:
-        return HTMLResponse(content="<h1>image-playground.html not found</h1>", status_code=404)
-    return HTMLResponse(content=html_content)
-   
+async def serve_vetra():
+    html = await get_github_file(FILES["html"])
+    css = await get_github_file(FILES["css"])
+    js = await get_github_file(FILES["js"])
+
+    if not html:
+        return HTMLResponse(content="<h1>index.html not found on GitHub</h1>", status_code=404)
+
+    final_html = html.replace(
+        "</head>",
+        f"<style>{css or '/* CSS not found */'}</style></head>"
+    ).replace(
+        "</body>",
+        f"<script>{js or '// JS not found'}</script></body>"
+    )
+
+    return HTMLResponse(content=final_html)
+
+
+
+
+
 # Model routes
 @app.get("/api/v1/models")
 @app.get("/models")