Upload main.py

main.py

@@ -117,6 +117,13 @@ trans = SyncTranslator()
 
 app = FastAPI(docs_url=None, redoc_url="/")
 
+contact_support = """
+We are aware that AI is currently offline. This seems to be caused by the API
+We are investigating and doing our best to get things back online as soon as possible. Thank you for your patience
+
+Contact Support @xtdevs
+"""
+
 def get_all_api_keys():
     user = collection.find({})
     api_keys = []
@@ -472,18 +479,12 @@ def ocr_space_url(
 
 @app.get("/ryuzaki/nochatgpt", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
 def no_chatgpt(query: str=None):
-    contact_support = """
-We are aware that AI is currently offline. This seems to be caused by the API
-We are investigating and doing our best to get things back online as soon as possible. Thank you for your patience
-
-Contact Support @xtdevs
-"""
     try:
         url = "http://216.98.10.228:5000"
         api_method = f"{url}/gpt?ask={query}"
         response = requests.get(api_method)
         if response.status_code != 200:
-            return SuccessResponse(status="False", randydev={"message": f"Error response"})
+            return SuccessResponse(status="False", randydev={"message": contact_support})
         return SuccessResponse(status="True", randydev={"message": response.text})
     except requests.exceptions.ConnectTimeout:
         return SuccessResponse(status="False", randydev={"message": contact_support})
@@ -494,27 +495,17 @@ Contact Support @xtdevs
 def free_chatgpt4_beta(item: ChatgptCustom):
     try:
         response = RendyDevChat(item.query).get_response_beta(joke=True)
-        return SuccessResponse(
-            status="True",
-            randydev={
-                "message": response
-            }
-        )
+        return SuccessResponse(status="True", randydev={"message": response})
     except:
-        return {"status": "false", "message": "Error response."}
+        return SuccessResponse(status="False", randydev={"message": contact_support})
 
 @app.get("/ryuzaki/freechatgpt-bing", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def free_chatgpt4_bing(item: ChatgptCustom):
     try:
         response = RendyDevChat(query).get_response_bing(bing=True)
-        return SuccessResponse(
-            status="True",
-            randydev={
-                "message": response
-            }
-        )
+        return SuccessResponse(status="True", randydev={"message": response})
     except:
-        return {"status": "false", "message": "Error response."}
+        return SuccessResponse(status="False", randydev={"message": contact_support})
 
 @app.post("/ryuzaki/ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def ryuzaki_ai(
@@ -528,16 +519,10 @@ def ryuzaki_ai(
             first_result = response_data[0]
             if "generated_text" in first_result:
                 message = first_result["generated_text"]
-                return SuccessResponse(
-                    status="True",
-                    randydev={
-                        "ryuzaki_text": message
-                    }
-                )
-        return {"status": "false", "message": "Invalid response format"}
-
+                return SuccessResponse(status="True", randydev={"ryuzaki_text": message})
+        return SuccessResponse(status="False", randydev={"message": contact_support})
     except Exception:
-        return {"status": "false", "message": "Internal server error"}
+        return SuccessResponse(status="False", randydev={"message": contact_support})
 
 @app.post("/ryuzaki/opendalle")
 def open_dalle(
@@ -565,7 +550,7 @@ def open_dalle(
     if encoded_string:
         return SuccessResponse(status="True", randydev={"data": encoded_string})
     else:
-        return SuccessResponse(status="False", randydev={"data": "Not found image data"})
+        return SuccessResponse(status="False", randydev={"data": contact_support})
 
 @app.post("/ryuzaki/picsart-pro")
 def Picsart_Pro(
@@ -607,7 +592,7 @@ def Picsart_Pro(
     if encoded_string:
         return SuccessResponse(status="True", randydev={"data": encoded_string})
     else:
-        return SuccessResponse(status="False", randydev={"data": "Not found image data"})
+        return SuccessResponse(status="False", randydev={"data": contact_support})
 
 @app.post("/ryuzaki/anime-styled")
 def Anime_Styled(
@@ -635,7 +620,7 @@ def Anime_Styled(
     if encoded_string:
         return SuccessResponse(status="True", randydev={"data": encoded_string})
     else:
-        return SuccessResponse(status="False", randydev={"data": "Not found image data"})
+        return SuccessResponse(status="False", randydev={"data": contact_support})
 
 @app.post("/ryuzaki/unsplash")
 def image_unsplash(item: GetImageUnsplash):
@@ -663,12 +648,6 @@ def image_unsplash(item: GetImageUnsplash):
 
 @app.post("/ryuzaki/chatgpt-model", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def chatgpt_model(item: ChatgptModel):
-    contact_support = """
-We are aware that AI is currently offline. This seems to be caused by the API
-We are investigating and doing our best to get things back online as soon as possible. Thank you for your patience
-
-Contact Support @xtdevs
-"""
     url = "https://lexica.qewertyy.me/models"
     if item.is_models:
         params = {"model_id": item.model_id, "prompt": item.query}
@@ -712,19 +691,16 @@ def chatgpt3_turbo(
                 }
             )
         except Exception:
-            return SuccessResponse(status="False", randydev={"message": f"Error responding"})
+            return SuccessResponse(status="False", randydev={"message": contact_support})
     else:
         url = "https://lexica.qewertyy.me/models"
         params = {"model_id": 5, "prompt": item.query}
         response = requests.post(url, params=params)
         if response.status_code != 200:
-            return f"Error status: {response.status_code}"
+            return SuccessResponse(status="False", randydev={"message": contact_support})
         check_response = response.json()
         answer = check_response.get("content")
-        return SuccessResponse(
-            status="True",
-            randydev={"message": answer}
-        )
+        return SuccessResponse(status="True", randydev={"message": answer})
 
 @app.post("/ryuzaki/chatgpt4-turbo", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def chatgpt4_turbo(
@@ -748,7 +724,7 @@ def chatgpt4_turbo(
             }
         )
     except Exception:
-        return SuccessResponse(status="False", randydev={"message": f"Error responding:"})
+        return SuccessResponse(status="False", randydev={"message": contact_support})
 
 @app.post("/ryuzaki/google-ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def v1beta3_google_ai(
@@ -770,14 +746,9 @@ def v1beta3_google_ai(
         answer = response_str["candidates"]
         for results in answer:
             message = results.get("output")
-        return SuccessResponse(
-            status="True",
-            randydev={
-                "message": message
-            }
-        )
+        return SuccessResponse(status="True", randydev={"message": message})
     except:
-        return {"status": "false", "message": "Error response."}
+        return SuccessResponse(status="False", randydev={"message": contact_support})
 
 @app.post("/ryuzaki/gemini-ai-pro", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def gemini_pro(item: GeminiPro):
@@ -787,12 +758,6 @@ def gemini_pro(item: GeminiPro):
 
     Today is {dt.now():%A %d %B %Y %H:%M}
     """
-    contact_support = """
-We are aware that AI is currently offline. This seems to be caused by the API
-We are investigating and doing our best to get things back online as soon as possible. Thank you for your patience
-
-Contact Support @xtdevs
-"""
     if item.is_multi_chat:
         selected_api_key = ASSISTANT_GOOGLE_API_KEYS or item.gemini_api_key
         try:
@@ -849,14 +814,9 @@ def v1beta2_google_ai(
         answer = response_str["candidates"]
         for results in answer:
             message = results.get("content")
-        return SuccessResponse(
-            status="True",
-            randydev={
-                "message": message
-            }
-        )
+        return SuccessResponse(status="True", randydev={"message": message})
     except:
-        return {"status": "false", "message": "Error response."}
+        return SuccessResponse(status="False", randydev={"message": contact_support})
 
 @app.post("/ryuzaki/new-monitor", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def new_monitor(
@@ -1008,24 +968,19 @@ def chatbot(item: ChatBots):
     except:
         return {"status": "false", "message": "Error response."}
 
-@app.get("/ryuzaki/llama", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+@app.get("/ryuzaki/llama", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
 def get_llama(item: ChatgptCustom):
     api_url = SOURCE_WHAT_GAY_URL
     params = {"query": item.query}
     x = requests.get(f"{api_url}/llama", params=params)
     if x.status_code != 200:
-        return "Error api request"
+        return SuccessResponse(status="False", randydev={"message": contact_support})
     try:
         y = x.json()
         response = y["answer"]
-        return {
-            "status": "true",
-            "randydev":{
-                "message": response
-            }
-        }
+        return SuccessResponse(status="True", randydev={"message": response})
     except:
-        return {"status": "false", "message": "Error response."}
+        return SuccessResponse(status="False", randydev={"message": contact_support})
 
 @app.get("/ryuzaki/waifu", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def waifu_pics(item: WaifuPics):