Update app.py

app.py

@@ -11,7 +11,7 @@ import cv2
 from PIL import Image
 import zipfile
 import io
-import tempfile
+import json
 import os
 
 DEVICE = Accelerator().device
@@ -85,13 +85,18 @@ def generate_image_embeddings(zip_file):
 
         embeddings = np.vstack(embeddings)
 
-        # Save embeddings to a temporary file
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".npy") as tmp:
-            np.save(tmp.name, embeddings)
-            output_path = tmp.name
+        # Create JSON output
+        result = json.dumps(
+            {
+                "embeddings": embeddings.tolist(),
+                "shape": list(embeddings.shape),
+                "count": len(images),
+            },
+            indent=2,
+        )
 
         message = f"✅ Successfully generated embeddings for {len(images)} images\nShape: {embeddings.shape}"
-        return output_path, message
+        return result, message
 
     except Exception as e:
         return None, f"❌ Error: {str(e)}"
@@ -170,13 +175,19 @@ def generate_video_embeddings(video_file, fps):
 
         embeddings = np.vstack(embeddings)
 
-        # Save embeddings to a temporary file
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".npy") as tmp:
-            np.save(tmp.name, embeddings)
-            output_path = tmp.name
+        # Create JSON output
+        result = json.dumps(
+            {
+                "embeddings": embeddings.tolist(),
+                "shape": list(embeddings.shape),
+                "count": len(frames),
+                "fps": fps,
+            },
+            indent=2,
+        )
 
         message = f"✅ Successfully generated embeddings for {len(frames)} frames (extracted at {fps} fps)\nShape: {embeddings.shape}"
-        return output_path, message
+        return result, message
 
     except Exception as e:
         return None, f"❌ Error: {str(e)}"
@@ -194,7 +205,7 @@ with gr.Blocks(title="Video & Image Embedding Generator") as demo:
                 zip_input = gr.File(label="Upload ZIP file", file_types=[".zip"])
                 img_submit_btn = gr.Button("Generate Embeddings", variant="primary")
             with gr.Column():
-                img_output = gr.File(label="Download Embeddings (.npy)")
+                img_output = gr.JSON(label="Embeddings (JSON)")
                 img_status = gr.Textbox(label="Status", lines=3)
 
         img_submit_btn.click(
@@ -219,7 +230,7 @@ with gr.Blocks(title="Video & Image Embedding Generator") as demo:
                 )
                 vid_submit_btn = gr.Button("Generate Embeddings", variant="primary")
             with gr.Column():
-                vid_output = gr.File(label="Download Embeddings (.npy)")
+                vid_output = gr.JSON(label="Embeddings (JSON)")
                 vid_status = gr.Textbox(label="Status", lines=3)
 
         vid_submit_btn.click(
@@ -233,8 +244,8 @@ with gr.Blocks(title="Video & Image Embedding Generator") as demo:
     ### 📝 Notes:
     - Images in ZIP: Supports PNG, JPG, JPEG, BMP, WEBP formats
     - Video: Supports common video formats (MP4, AVI, MOV, etc.)
-    - Output: NumPy array file (.npy) containing normalized embeddings
-    - Load embeddings: `embeddings = np.load('embeddings.npy')`
+    - Output: JSON object containing normalized embeddings with metadata
+    - Structure: `{"embeddings": [...], "shape": [n, dim], "count": n, "fps": f}`
     """
     )