Update app.py

app.py

@@ -3,7 +3,9 @@ from huggingface_hub import HfApi
 import os
 import uuid
 from fastapi import FastAPI, UploadFile, File
+from fastapi.middleware.cors import CORSMiddleware
 import uvicorn
+import shutil
 
 # --- SYSTEM CONFIGURATION ---
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -12,26 +14,39 @@ HIDDEN_PATH = "data/blobs/0x44"
 
 # Initialize Connection
 api = HfApi(token=HF_TOKEN)
+
+# Initialize FastAPI
 app = FastAPI()
 
+# Add CORS to allow mobile app requests
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 # --- HELPER FUNCTION ---
-def upload_logic(file_obj, filename_hint="video.mp4"):
+def upload_logic(file_path, filename_hint="video.mp4"):
     """
     Core upload logic shared by both Gradio (Web) and FastAPI (Mobile)
     """
     try:
         # 1. Generate Stealth Name
         request_id = str(uuid.uuid4())
-        extension = os.path.splitext(filename_hint)[1] or ".bin"
-        if extension == ".bin": extension = ".mp4" # Default to mp4 if unknown
-        
+        extension = os.path.splitext(filename_hint)[1] or ".mp4"
+        # Ensure we don't end up with weird extensions
+        if len(extension) > 5 or extension == ".bin": 
+            extension = ".mp4"
+            
         shard_name = f"sys_core_dump_{request_id}{extension}"
         
         print(f"⚙️ Allocating Shard: {shard_name}...")
 
         # 2. Upload to Hidden Folder
         api.upload_file(
-            path_or_fileobj=file_obj,
+            path_or_fileobj=file_path,
             path_in_repo=f"{HIDDEN_PATH}/{shard_name}",
             repo_id=DATASET_ID,
             repo_type="dataset",
@@ -49,8 +64,7 @@ def upload_logic(file_obj, filename_hint="video.mp4"):
 # --- WEB INTERFACE (GRADIO) ---
 def system_shard_allocator(packet_file):
     if packet_file is None: return "ERROR: No packet received."
-    
-    # Gradio passes a NamedTemporaryFile wrapper, we use its path
+    # Gradio handles cleanup, we just use the path
     url = upload_logic(packet_file.name, packet_file.name)
     return url if url else "ALLOCATION_ERROR"
 
@@ -61,11 +75,10 @@ with gr.Blocks(title="System Sync Daemon") as system_daemon:
         output_log = gr.Textbox(label="Allocation Path")
     
     btn = gr.Button("Execute Sync")
-    # Web uses 'predict' endpoint via Gradio Client
+    # 'predict' API name enabled for website compatibility
     btn.click(fn=system_shard_allocator, inputs=input_packet, outputs=output_log, api_name="predict")
 
 # --- MOBILE INTERFACE (FASTAPI) ---
-# This matches the endpoint in your mobile app: /upload_stream
 @app.post("/upload_stream")
 async def upload_stream(video: UploadFile = File(...)):
     """
@@ -73,17 +86,35 @@ async def upload_stream(video: UploadFile = File(...)):
     """
     print(f"📱 Receiving mobile stream: {video.filename}")
     
-    # Pass the specialized file object from FastAPI directly to HfApi
-    url = upload_logic(video.file, video.filename)
-    
-    if url:
-        return {"url": url}
-    else:
-        return {"error": "Upload failed"}, 500
+    try:
+        # Create a temporary file to store the upload
+        temp_filename = f"temp_{uuid.uuid4()}_{video.filename}"
+        with open(temp_filename, "wb") as buffer:
+            shutil.copyfileobj(video.file, buffer)
+            
+        # Upload
+        url = upload_logic(temp_filename, video.filename)
+        
+        # Cleanup
+        if os.path.exists(temp_filename):
+            os.remove(temp_filename)
+            
+        if url:
+            return {"url": url}
+        else:
+            return {"error": "Upload failed"}, 500
+            
+    except Exception as e:
+        print(f"Error processing mobile upload: {e}")
+        return {"error": str(e)}, 500
+
+# IMPORTANT: Enable queueing BEFORE mounting
+system_daemon.queue()
 
 # Mount Gradio to the root of FastAPI
+# This allows both / (Gradio) and /upload_stream (API) to work
 app = gr.mount_gradio_app(app, system_daemon, path="/")
 
-# If running locally (Space handles this automatically usually, but good for testing)
+# Entry point for HF Spaces
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)