Update main.py

main.py

@@ -9,11 +9,11 @@ from collections import defaultdict
 import secrets
 from config import DOCS_USERNAME, DOCS_PASSWORD
 
-# Initialize security
+# Initialize security for docs only
 security = HTTPBasic()
 
 def verify_credentials(credentials: HTTPBasicCredentials = Depends(security)):
-    """Verify HTTP Basic Auth credentials"""
+    """Verify HTTP Basic Auth credentials for docs only"""
     is_username_correct = secrets.compare_digest(credentials.username, DOCS_USERNAME)
     is_password_correct = secrets.compare_digest(credentials.password, DOCS_PASSWORD)
     
@@ -25,17 +25,37 @@ def verify_credentials(credentials: HTTPBasicCredentials = Depends(security)):
         )
     return credentials
 
-# Initialize FastAPI with auth for docs
+# Initialize FastAPI with auth for docs only
 app = FastAPI(
     title="File Sharing API",
     description="API for file sharing service",
     version="1.0.0",
-    docs_url="/docs",
-    redoc_url="/redoc",
-    dependencies=[Depends(verify_credentials)],
-    openapi_url="/openapi.json"
+    docs_url=None,  # Disable default docs
+    redoc_url=None  # Disable default redoc
 )
 
+# Create protected docs routes
+@app.get("/docs", include_in_schema=False)
+async def get_documentation(credentials: HTTPBasicCredentials = Depends(verify_credentials)):
+    from fastapi.openapi.docs import get_swagger_ui_html
+    return get_swagger_ui_html(openapi_url="/openapi.json", title="API Documentation")
+
+@app.get("/redoc", include_in_schema=False)
+async def get_redoc(credentials: HTTPBasicCredentials = Depends(verify_credentials)):
+    from fastapi.openapi.docs import get_redoc_html
+    return get_redoc_html(openapi_url="/openapi.json", title="API Documentation")
+
+@app.get("/openapi.json", include_in_schema=False)
+async def get_openapi(credentials: HTTPBasicCredentials = Depends(verify_credentials)):
+    from fastapi.openapi.utils import get_openapi
+    openapi_schema = get_openapi(
+        title="File Sharing API",
+        version="1.0.0",
+        description="API for file sharing service",
+        routes=app.routes
+    )
+    return openapi_schema
+
 # Create uploads directory if it doesn't exist
 UPLOAD_DIR = "uploads"
 if not os.path.exists(UPLOAD_DIR):