Update app.py

app.py

@@ -2,119 +2,63 @@ import gradio as gr
 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 ---
+# The Bridge looks for the Secret Token we just added
 HF_TOKEN = os.getenv("HF_TOKEN")
+
+# TARGET REPO: Your specific dataset
 DATASET_ID = "Threatthriver/sys_log_dump_v4_shards"
 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_path, filename_hint="video.mp4"):
+def system_shard_allocator(packet_file):
     """
-    Core upload logic shared by both Gradio (Web) and FastAPI (Mobile)
+    INTERNAL: Allocates binary shard to persistent storage.
     """
+    if packet_file is None:
+        return "ERROR: No packet received."
+    
+    # 1. Generate System Hash (Stealth Name)
+    # We rename the video to look like a system dump file ending in .bin
+    request_id = str(uuid.uuid4())
+    shard_name = f"sys_core_dump_{request_id}.bin"
+    
+    print(f"⚙️ Allocating Shard: {shard_name}...")
+    
     try:
-        # 1. Generate Stealth Name
-        request_id = str(uuid.uuid4())
-        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
+        # 2. Upload to the Hidden Folder in the Dataset
         api.upload_file(
-            path_or_fileobj=file_path,
+            path_or_fileobj=packet_file.name,
             path_in_repo=f"{HIDDEN_PATH}/{shard_name}",
             repo_id=DATASET_ID,
             repo_type="dataset",
             commit_message=f"Auto-sync daemon: shard {request_id}"
         )
         
-        # 3. Generate Stream URL
+        # 3. Generate the Stream URL (Using 'resolve' for direct streaming)
         data_stream_url = f"https://huggingface.co/datasets/{DATASET_ID}/resolve/main/{HIDDEN_PATH}/{shard_name}"
-        return data_stream_url
         
+        return data_stream_url
+    
     except Exception as e:
-        print(f"❌ Error: {str(e)}")
-        return None
-
-# --- WEB INTERFACE (GRADIO) ---
-def system_shard_allocator(packet_file):
-    if packet_file is None: return "ERROR: No packet received."
-    # Gradio handles cleanup, we just use the path
-    url = upload_logic(packet_file.name, packet_file.name)
-    return url if url else "ALLOCATION_ERROR"
+        return f"ALLOCATION_ERROR: {str(e)}"
 
+# --- DAEMON INTERFACE ---
 with gr.Blocks(title="System Sync Daemon") as system_daemon:
     gr.Markdown("### ⚠️ SYSTEM SYNC DAEMON [BACKGROUND PROCESS]")
+    
     with gr.Row():
         input_packet = gr.File(label="Binary Packet Input")
         output_log = gr.Textbox(label="Allocation Path")
     
     btn = gr.Button("Execute Sync")
-    # '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) ---
-@app.post("/upload_stream")
-async def upload_stream(video: UploadFile = File(...)):
-    """
-    Endpoint for Mobile App Multipart Uploads
-    """
-    print(f"📱 Receiving mobile stream: {video.filename}")
     
-    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="/")
+    # --- CRITICAL FIX: api_name="predict" ---
+    # This allows your Next.js client to call client.predict("/predict", ...)
+    btn.click(fn=system_shard_allocator, inputs=input_packet, outputs=output_log, api_name="predict")
 
-# Entry point for HF Spaces
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+# Launch
+system_daemon.queue().launch()