Upload main.py

main.py

@@ -83,6 +83,7 @@ SOURCE_ALPHA_URL = os.environ["SOURCE_ALPHA_URL"]
 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"]
 
 client_mongo = MongoClient(MONGO_URL)
 db = client_mongo["tiktokbot"]
@@ -381,6 +382,19 @@ async def get_image_unsplash(query: str, size: str="500x500"):
 
     return StreamingResponse(BytesIO(response.content), media_type="image/jpeg")
 
+@app.get("/ryuzaki/calligraphy")
+async def get_image_calligraphy(query: str):
+    url = SOURCE_CALLI_GRAPHY_URL
+    image_url = f"{url}?style=1&text={query}"
+
+    try:
+        response = requests.get(image_url)
+        response.raise_for_status()
+    except requests.exceptions.RequestException as e:
+        raise HTTPException(status_code=500, detail=f"Error fetching image: {e}")
+
+    return StreamingResponse(BytesIO(response.content), media_type="image/jpeg")
+
 @app.get("/ryuzaki/reverse")
 def google_reverse(
     engine: str="google_reverse_image",