Update app.py

app.py

@@ -1,8 +1,13 @@
+import g4f.api
+import g4f.Provider
 from fastapi import FastAPI, HTTPException, Depends, Header, Request
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 from typing import List
+from __future__ import annotations
+from g4f.typing import Messages, AsyncResult
 from g4f import ChatCompletion
+from g4f.Provider import BackendApi
 
 app = FastAPI()
 
@@ -11,9 +16,37 @@ models = [
     "gpt-4o", "gpt-4o-mini", "gpt-4",
     "gpt-4-turbo", "gpt-3.5-turbo",
     "claude-3.7-sonnet", "o3-mini", "o1", "grok-3", "gemini-2.5-pro-exp-03-25", "claude-3.5", 
-    "llama-3.1-405b"
+    "llama-3.1-405b", "deepseek-r1", *g4f.Provider.OpenaiAccount.get_models(), "flux", "flux-pro"
 ]
 
+url = "https://ahe.hopto.org"
+headers = {"Authorization": "Basic Z2dnOmc0Zl8="}
+
+BackendApi.working = True
+BackendApi.ssl = False
+BackendApi.url = url
+BackendApi.headers = headers
+
+class BackendApi(BackendApi):
+    working = True
+    ssl = False
+    url = url
+    headers = headers
+    image_models = ["flux", "flux-pro"]
+    models = ["deepseek-r1", *g4f.Provider.OpenaiAccount.get_models(), "flux", "flux-pro"]
+
+    @classmethod
+    async def create_async_generator(
+        cls,
+        model: str,
+        messages: Messages,
+        **kwargs
+    ) -> AsyncResult:
+        if model in g4f.Provider.OpenaiAccount.get_models():
+            kwargs["provider"] = "OpenaiAccount"
+        async for chunk in super().create_async_generator(model, messages, **kwargs):
+            yield chunk
+    
 # Request model
 class Message(BaseModel):
     role: str