Spaces:

pagezyhf
/

inference-cache-explorer

Sleeping

App Files Files Community

pagezyhf HF Staff commited on Mar 31

Commit

05e2837

1 Parent(s): bdc22f6

tesst

Browse files

Files changed (2) hide show

app/main.py +27 -3
app/templates/index.html +2 -2

app/main.py CHANGED Viewed

@@ -2,6 +2,7 @@ from fastapi import FastAPI, Request
 from fastapi.templating import Jinja2Templates
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import JSONResponse, FileResponse
 from optimum.neuron import utils
 import logging
 import sys
@@ -19,6 +20,15 @@ logger = logging.getLogger(__name__)
 app = FastAPI()
 # Get the absolute path to the static directory
 static_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "static")
 logger.info(f"Static directory path: {static_dir}")
@@ -39,7 +49,15 @@ async def health_check():
 @app.get("/")
 async def home(request: Request):
     logger.info("Home page requested")
-    return templates.TemplateResponse("index.html", {"request": request})
 @app.get("/api/models")
 async def get_model_list():
@@ -95,9 +113,15 @@ async def get_model_info_endpoint(model_id: str):
         )
 @app.get("/static/{path:path}")
-async def static_files(path: str):
     logger.info(f"Static file requested: {path}")
     file_path = os.path.join(static_dir, path)
     if os.path.exists(file_path):
-        return FileResponse(file_path)
     return JSONResponse(status_code=404, content={"error": "File not found"})

 from fastapi.templating import Jinja2Templates
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import JSONResponse, FileResponse
+from fastapi.middleware.cors import CORSMiddleware
 from optimum.neuron import utils
 import logging
 import sys
 app = FastAPI()
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 # Get the absolute path to the static directory
 static_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "static")
 logger.info(f"Static directory path: {static_dir}")
 @app.get("/")
 async def home(request: Request):
     logger.info("Home page requested")
+    # Get the base URL with proper protocol
+    base_url = str(request.base_url).replace("http://", "https://")
+    return templates.TemplateResponse(
+        "index.html",
+        {
+            "request": request,
+            "base_url": base_url
+        }
+    )
 @app.get("/api/models")
 async def get_model_list():
         )
 @app.get("/static/{path:path}")
+async def static_files(path: str, request: Request):
     logger.info(f"Static file requested: {path}")
     file_path = os.path.join(static_dir, path)
     if os.path.exists(file_path):
+        response = FileResponse(file_path)
+        # Ensure proper content type
+        if path.endswith('.css'):
+            response.headers["content-type"] = "text/css"
+        elif path.endswith('.js'):
+            response.headers["content-type"] = "application/javascript"
+        return response
     return JSONResponse(status_code=404, content={"error": "File not found"})

app/templates/index.html CHANGED Viewed

@@ -5,7 +5,7 @@
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>Model Configuration Explorer</title>
     <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css">
-    <link rel="stylesheet" href="{{ url_for('static', path='/css/styles.css') }}">
 </head>
 <body>
     <div class="container">
@@ -60,6 +60,6 @@
         </section>
     </div>
-    <script src="{{ url_for('static', path='/js/main.js') }}"></script>
 </body>
 </html>

     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>Model Configuration Explorer</title>
     <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css">
+    <link rel="stylesheet" href="{{ base_url }}static/css/styles.css">
 </head>
 <body>
     <div class="container">
         </section>
     </div>
+    <script src="{{ base_url }}static/js/main.js"></script>
 </body>
 </html>