refactor: Implement lazy loading for images in DashboardV2 and update…

… chat route

app.py

@@ -136,6 +136,7 @@ class ChatByIdHistory(BaseModel):
     model: str
 
 class SubscriptionRequest(BaseModel):
+    """Subscription request model for the subscription endpoint."""
     redirect_url: str
 
 model_company_mapping = {
@@ -151,6 +152,10 @@ class SubscriptionRequest(BaseModel):
         "model" : ChatOpenAI,
         "premium" : True,
     },
+    "gpt-4o-mini" : {
+        "model" : ChatOpenAI,
+        "premium" : False,
+    },
     "gpt-4o": {
         "model" : ChatOpenAI,
         "premium" : True,
@@ -167,6 +172,10 @@ class SubscriptionRequest(BaseModel):
         "model" : ChatAnthropic,
         "premium" : False,
     },
+    "claude-3-5-sonnet-20240620" : {
+        "model" : ChatAnthropic,
+        "premium" : True,
+    },
     "mistral-tiny-2312": {
         "model" : ChatMistralAI,
         "premium" : False,
@@ -239,6 +248,18 @@ class SubscriptionRequest(BaseModel):
         "model" : ChatTogether,
         "premium" : True,
     },
+    "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo" : {
+        "model" : ChatTogether,
+        "premium" : True,
+    },
+    "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo" : {
+        "model" : ChatTogether,
+        "premium" : True,
+    },
+    "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo" : {
+        "model" : ChatTogether,
+        "premium" : True,
+    },
     "google/gemma-2b-it": {
         "model" : ChatTogether,
         "premium" : False,
@@ -846,7 +867,6 @@ async def is_user_subscribed(token_info: dict = Depends(verify_token)):
         raise HTTPException(status_code=500, detail="Internal server error") from e
 
 if __name__ == "__main__":
-    import uvicorn
 
     # Use multiprocessing for parallel request handling
     uvicorn.run(app, host="0.0.0.0", port=5000, log_level="info")

web/src/pages/DashboardV2.js

@@ -35,7 +35,7 @@ export const DashboardV2 = () => {
     const [isLoadingGeneratingChat, setIsLoadingGeneratingChat] = useState(false);
     const accessToken = localStorage.getItem("accessToken");
     const [modal, setModal] = useState(null);
-    const [isSidebarOpen, setIsSidebarOpen] = useState(true);
+    const [isSidebarOpen, setIsSidebarOpen] = useState(false);
     const [selectedModel, setSelectedModel] = useState(modelOptions[0]);
     const [messages, setMessages] = useState([]);
     let previousModel = null;