Upload folder using huggingface_hub

database.py

@@ -462,3 +462,26 @@ def update_job_status(job_id: str, status: str, result: Optional[Dict] = None, e
         return False
 
 
+
+def get_latest_job(username: str) -> Optional[Dict[str, Any]]:
+    """Retrieve the most recent job for a user."""
+    conn = get_db_connection()
+    c = conn.cursor()
+    c.execute('''
+        SELECT * FROM jobs 
+        WHERE username = ? 
+        ORDER BY created_at DESC 
+        LIMIT 1
+    ''', (username,))
+    row = c.fetchone()
+    conn.close()
+    
+    if row:
+        job = dict(row)
+        if job['result']:
+            try:
+                job['result'] = json.loads(job['result'])
+            except:
+                job['result'] = None
+        return job
+    return None

main.py

@@ -1543,15 +1543,39 @@ async def analyze_image(
     
     # Enqueue Worker (Pass ID, not bytes)
     background_tasks.add_task(process_analysis_job, task_id, request.image_id, current_user.username)
-    
-    logger.info(f"Job Created: {task_id} for Image: {request.image_id}")
-    
-    return {
         "task_id": task_id, 
         "status": "queued",
         "image_id": request.image_id
     }
 
+@app.get("/job/current")
+async def get_current_job(current_user: User = Depends(get_current_active_user)):
+    """
+    Get the latest job state for the user to restore UI on refresh.
+    Returns 404 if no recent job found (< 24h).
+    """
+    job = database.get_latest_job(current_user.username)
+    if not job:
+        raise HTTPException(status_code=404, detail="No active job")
+        
+    # Check if job is stale (e.g. > 24 hours old)
+    # If completed and old, we might not want to auto-load it on fresh login
+    # But for F5 refresh, we definitely want it.
+    # Heuristic: If < 1 hour, always return.
+    
+    created_at = job.get('created_at', 0)
+    if time.time() - created_at > 86400: # 24 hours
+         raise HTTPException(status_code=404, detail="Job expired")
+         
+    return {
+        "task_id": job['id'],
+        "status": job['status'],
+        "result": job['result'],
+        "error": job.get('error'),
+        "created_at": created_at,
+        "image_id": job.get('storage_path')
+    }
+
 @app.get("/result/{task_id}")
 async def get_result(task_id: str, current_user: User = Depends(get_current_user)):
     """