Upload main.py

main.py

@@ -490,21 +490,34 @@ def chatgpt4_support(
         return {"status": "false", "message": "Error response."}
 
 @app.post("/ryuzaki/chatgpt-model")
-def chatgpt_model(
-    query: str=None,
-    model_id: int=1,
-    is_models: bool=True
-):
-    try:
-        response = RendyDevChat(query).get_response_model(model_id=model_id, is_models=is_models)
+def chatgpt_model(query: str=None, model_id: int=None, is_models: bool=False):
+    url = "https://lexica.qewertyy.me/models"
+    if is_models:
+        params = {"model_id": model_id, "prompt": query}
+        response = requests.post(url, params=params)
+        if response.status_code != 200:
+            return f"Error status: {response.status_code}"
+        check_response = response.json()
+        answer = check_response.get("content")
         return {
             "status": "true",
             "randydev":{
-                "message": response
+                "message": answer
+            }
+        }
+    else:
+        params = {"model_id": 5, "prompt": query}
+        response = requests.post(url, params=params)
+        if response.status_code != 200:
+            return f"Error status: {response.status_code}"
+        check_response = response.json()
+        answer = check_response.get("content")
+        return {
+            "status": "true",
+            "randydev":{
+                "message": answer
             }
         }
-    except:
-        return {"status": "false", "message": "Error response."}
 
 @app.get("/ryuzaki/freechatgpt-beta")
 def free_chatgpt4_beta(query: str):