Disable access log

Dockerfile

@@ -12,4 +12,4 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=user . /app
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--no-access-log"]

app.py

@@ -23,6 +23,8 @@ from g4f.providers.response import Parameters, FinishReason, Usage, Reasoning
 from g4f.errors import ModelNotSupportedError
 from g4f import debug
 
+import demo
+
 class BackendApi(AsyncGeneratorProvider, ProviderModelMixin):
     url = "https://ahe.hopto.org"
     working = True
@@ -84,7 +86,7 @@ class BackendApi(AsyncGeneratorProvider, ProviderModelMixin):
 
         async with StreamSession(
             proxy=proxy,
-            headers={"Accept": "text/event-stream"},
+            headers={"Accept": "text/event-stream", **demo.headers},
             timeout=timeout
         ) as session:
             model = cls.get_model(model)
@@ -215,15 +217,6 @@ def create_app():
     g4f.api.AppConfig.gui = True
     g4f.api.AppConfig.demo = False
 
-    class NoHomeFilter(logging.Filter):
-        def filter(self, record):
-            if '"GET / HTTP/1.1" 200 OK' in record.getMessage():
-                return False
-            if '"GET /static/' in record.getMessage():
-                return False
-            return True
-    logging.getLogger("uvicorn.access").addFilter(NoHomeFilter())
-
     app = FastAPI()
 
     # Add CORS middleware
@@ -241,11 +234,12 @@ def create_app():
     api.register_authorization()
     api.register_validation_exception_handler()
 
-    headers = {}
-
     @app.get("/download/{filename}", response_class=RedirectResponse)
     async def download(filename, request: Request):
         filename = os.path.basename(filename)
+        if "." not in filename:
+            target = os.path.join(images_dir, filename)
+            filename = f"{filename}.jpg"
         target = os.path.join(images_dir, filename)
         if not os.path.exists(target):
             url = str(request.query_params).split("url=", 1)[1]
@@ -255,9 +249,8 @@ def create_app():
                     [source_url],
                     target=target,
                     ssl=False,
-                    headers=headers if source_url.startswith(BackendApi.url) else None)
+                    headers=demo.headers if source_url.startswith(BackendApi.url) else None)
             if not os.path.exists(target):
-        
                 return Response(status_code=404)
         return RedirectResponse(f"/images/{filename}")
 
@@ -266,4 +259,14 @@ def create_app():
 
     return app
 
+
+class NoHomeFilter(logging.Filter):
+    def filter(self, record):
+        if '"GET / HTTP/1.1" 200 OK' in record.getMessage():
+            return False
+        if '"GET /static/' in record.getMessage():
+            return False
+        return True
+logging.getLogger("uvicorn.access").addFilter(NoHomeFilter())
+
 app = create_app()

clean.py

@@ -0,0 +1,5 @@
+import os
+for root, _, files in os.walk("./generated_images"):
+    for file in files:
+        if "." not in file:
+            os.rename(file, f"./generated_images/{file}.jpg")

demo/__init__.py

@@ -0,0 +1 @@
+headers = {"Authorization": "Basic Z2dnOmc0Zl8="}