image serving from cloudinary

app.py

@@ -16,10 +16,12 @@ import utils as U
 from helpers.auth import runWithAuth
 from helpers.sidebar import showSidebar
 from helpers.activities import saveLatestActivity
+from helpers.imageCdn import getCdnUrl
 
 from dotenv import load_dotenv
 load_dotenv()
 
+
 ModelType = Literal["GPT4", "CLAUDE", "LLAMA"]
 ModelConfig = TypedDict("ModelConfig", {
     "client": OpenAI | Groq | anthropic.Anthropic,
@@ -465,6 +467,8 @@ def mainApp():
 
             imageContainer = st.empty()
             imagePath = __paintImageIfApplicable(imageContainer, prompt, response)
+            if imagePath:
+                imagePath = getCdnUrl(imagePath)
 
             st.session_state.chatHistory.append({
                 "role": "assistant",

helpers/imageCdn.py

@@ -0,0 +1,22 @@
+import cloudinary
+from cloudinary import CloudinaryImage
+import cloudinary.uploader
+import cloudinary.api
+import utils as U
+
+config = cloudinary.config(secure=True)
+
+
+def __getImageId(imagePath: str):
+    imagePath = imagePath.replace('\\', '/')
+    parts = imagePath.split('/')
+    return parts[-2]
+
+
+def getCdnUrl(imagePath: str) -> str:
+    imageId = __getImageId(imagePath)
+    cloudinary.uploader.upload(imagePath, public_id=imageId, unique_filename=False, overwrite=True)
+    U.pprint(f"Image uploaded to CDN: {imageId}")
+    srcURL = CloudinaryImage(imageId).build_url()
+    U.pprint(f"Image CDN URL: {srcURL}")
+    return srcURL

requirements.txt

@@ -6,3 +6,5 @@ gradio_client
 anthropic
 supabase
 descope
+cloudinary
+