Update app.py

app.py

@@ -1,5 +1,8 @@
 import gradio as gr
-
+import tempfile
+import uuid
+import os
+import base64
 from nivra_agent import nivra_chat, nivra_vision
 from indicTrans3Infer import translate_text
 
@@ -16,8 +19,19 @@ def chat_fn(message, history):
 # ===============================
 # Translation endpoint
 # ===============================
-def indictrans_translate(text: str, target_language: str):
-    return translate_text(text, target_language)
+UPLOAD_DIR = "/tmp/uploads"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+
+def upload_image(image_base64: str):
+    img_bytes = base64.b64decode(image_base64)
+    filename = f"{uuid.uuid4()}.png"
+    path = os.path.join(UPLOAD_DIR, filename)
+
+    with open(path, "wb") as f:
+        f.write(img_bytes)
+
+    # Expose via HF Space static file serving
+    return f"{os.environ['SPACE_HOST']}/file={path}"
 
 # ===============================
 # 🆕 Vision endpoint (CRITICAL FIX)