Upload main.py

main.py

@@ -84,6 +84,7 @@ SOURCE_WAIFU_URL = os.environ["SOURCE_WAIFU_URL"]
 SOURCE_TIKTOK_WTF_URL = os.environ["SOURCE_TIKTOK_WTF_URL"]
 SOURCE_TIKTOK_TECH_URL = os.environ["SOURCE_TIKTOK_TECH_URL"]
 SOURCE_CALLI_GRAPHY_URL = os.environ["SOURCE_CALLI_GRAPHY_URL"]
+SOURCE_WHAT_GAY_URL = os.environ["SOURCE_WHAT_GAY_URL"]
 
 client_mongo = MongoClient(MONGO_URL)
 db = client_mongo["tiktokbot"]
@@ -586,6 +587,25 @@ def chatbot(
     except:
         return {"status": "false", "message": "Error response."}
 
+@app.get("/ryuzaki/llama")
+def get_llama(query: str):
+    api_url = SOURCE_WHAT_GAY_URL
+    params = {"query": query}
+    x = requests.get(f"{api_url}/llama", params=params)
+    if x.status_code != 200:
+        return "Error api request"
+    try:
+        y = x.json()
+        response = y["answer"]
+        return {
+            "status": "true",
+            "randydev":{
+                "message": response
+            }
+        }
+    except:
+        return {"status": "false", "message": "Error response."}
+
 @app.get("/ryuzaki/waifu")
 def waifu_pics(
     types: str="sfw",