Middleware and Proxy Fix

fastapi_app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, Request, HTTPException, Response, Body
 from fastapi.responses import JSONResponse, HTMLResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
+from starlette.staticfiles import StaticFiles as StarletteStaticFiles
 import os
 import base64
 import numpy as np
@@ -14,13 +15,35 @@ import logging
 from pydantic import BaseModel
 from typing import Optional, Dict, Any, List
 from topic_segmenter import TopicSegmenter
+from middleware import HTTPSProxyMiddleware
 
-app = FastAPI(title="Flowify", description="Audio transcription and topic segmentation API")
+# Configure logging
 logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+app = FastAPI(title="Flowify", description="Audio transcription and topic segmentation API")
+
+# Add custom middleware for HTTPS handling
+app.add_middleware(HTTPSProxyMiddleware)
 
-# Configure templates and static files
+# Configure templates with HTTPS handling
 templates = Jinja2Templates(directory="templates")
-app.mount("/static", StaticFiles(directory="static"), name="static")
+
+# Custom static file handling to ensure proper URL scheme
+class SecureStaticFiles(StarletteStaticFiles):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+    
+    async def get_response(self, path, scope):
+        # Force scope to use HTTPS when behind proxy
+        if "headers" in scope and any(h[0] == b"x-forwarded-proto" and h[1] == b"https" for h in scope["headers"]):
+            scope["scheme"] = "https"
+        
+        response = await super().get_response(path, scope)
+        return response
+
+# Mount static files with secure handler
+app.mount("/static", SecureStaticFiles(directory="static"), name="static")
 
 # Models and locks
 models = {}
@@ -38,7 +61,12 @@ class ModelCheck(BaseModel):
 
 @app.get("/", response_class=HTMLResponse)
 async def index(request: Request):
-    return templates.TemplateResponse("index.html", {"request": request})
+    logger.info(f"Serving index.html with base_url={request.base_url}, url={request.url}")
+    
+    # Ensure template knows the correct scheme
+    context = {"request": request}
+    
+    return templates.TemplateResponse("index.html", context)
 
 def get_model(model_name):
     """Load and cache the model"""

middleware.py

@@ -0,0 +1,32 @@
+from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.responses import RedirectResponse
+from fastapi import Request
+import logging
+
+logger = logging.getLogger(__name__)
+
+class HTTPSProxyMiddleware(BaseHTTPMiddleware):
+    """Middleware to handle HTTPS for proxied applications like Hugging Face Spaces."""
+    
+    async def dispatch(self, request: Request, call_next):
+        """
+        Dispatches the middleware.
+        
+        1. Modifies the request headers to ensure proper URL scheme
+        2. Logs details about the request for debugging
+        """
+        # Log original request info for debugging
+        logger.info(f"Original request: scheme={request.url.scheme}, url={request.url}")
+        
+        # Detect if we're behind a proxy that's forwarding HTTPS as HTTP
+        forwarded_proto = request.headers.get("X-Forwarded-Proto")
+        
+        # If we have a forwarded proto header and it's HTTPS, 
+        # modify the request's scope to ensure URLs are generated with HTTPS
+        if forwarded_proto == "https" and request.url.scheme == "http":
+            logger.info("Detected HTTPS behind proxy, updating request scope")
+            request.scope["scheme"] = "https"
+        
+        # Continue processing the request
+        response = await call_next(request)
+        return response 

templates/index.html

@@ -5,9 +5,13 @@
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>Flowify</title>
     
-    <link rel="stylesheet" href="{{ url_for('static', path='styles.css') }}">
+    <!-- Force HTTPS for the static assets -->
+    <link rel="stylesheet" href="{{ request.url_for('static', path='styles.css') }}">
     <link type="text/css" rel="stylesheet" href="https://unpkg.com/jsmind@0.8.6/style/jsmind.css" />
     <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.4/css/all.min.css">
+    
+    <!-- Add base tag to ensure all relative URLs use the correct scheme -->
+    <base href="{{ request.url }}">
 </head>
 <body>
     <div class="app-container">