Update app.py

app.py

@@ -330,37 +330,45 @@ async def list_files():
 async def get_state():
     """
     Get the current application state (including file processing status).
-    This endpoint now ensures it fetches the latest state from Hugging Face.
+    This endpoint now ensures it fetches the latest state from the local file.
     """
-    # 1. Load the file_states from the local file, which is updated by client uploads.
-    # We assume the local file exists because it's either downloaded on startup
-    # or created by a client upload.
-    if not os.path.exists(STATE_FILE_PATH):
-        # If the file is missing, attempt a download from HF as a fallback
-        download_state_file_from_hf()
-        if not os.path.exists(STATE_FILE_PATH):
-            # If still not found (e.g., first run and 404 on HF), return default
-            return {"state": {"next_download_index": 0, "file_states": {}}}
-        
-    try:
-        with open(STATE_FILE_PATH, "r") as f:
-            state_content = json.load(f)
-            
-            # Ensure the required keys for the client are present
-            if "file_states" not in state_content:
-                state_content["file_states"] = {}
-            if "next_download_index" not in state_content:
-                state_content["next_download_index"] = 0
+    # 1. Always read the latest state from the local file (which gets updated by client uploads)
+    if os.path.exists(STATE_FILE_PATH):
+        try:
+            with open(STATE_FILE_PATH, "r") as f:
+                state_content = json.load(f)
                 
-            # Merge the internal app_state counters (which are not critical for locking)
-            # into the returned state for completeness, but the core file_states come from the file.
-            state_content.update(app_state)
-            
-            return {"state": state_content}
-            
-    except Exception as e:
-        print(f"Error reading local state file for /state/ endpoint: {e}")
-        raise HTTPException(status_code=500, detail="Error processing state file.")
+                # Ensure the required keys for the client are present
+                if "file_states" not in state_content:
+                    state_content["file_states"] = {}
+                if "next_download_index" not in state_content:
+                    state_content["next_download_index"] = 0
+                
+                print(f"✅ Read state from file: next_download_index={state_content.get('next_download_index')}, file_states={len(state_content.get('file_states', {}))}")
+                return {"state": state_content}
+                
+        except Exception as e:
+            print(f"Error reading local state file for /state/ endpoint: {e}")
+            raise HTTPException(status_code=500, detail="Error processing state file.")
+    
+    # 2. If file doesn't exist, try to download from HF as fallback
+    download_state_file_from_hf()
+    if os.path.exists(STATE_FILE_PATH):
+        try:
+            with open(STATE_FILE_PATH, "r") as f:
+                state_content = json.load(f)
+                
+                if "file_states" not in state_content:
+                    state_content["file_states"] = {}
+                if "next_download_index" not in state_content:
+                    state_content["next_download_index"] = 0
+                    
+                return {"state": state_content}
+        except Exception as e:
+            print(f"Error reading downloaded state file: {e}")
+    
+    # 3. Return default state if file still not found
+    return {"state": {"next_download_index": 0, "file_states": {}}}
 
 # --- Main execution block for testing/running ---
 if __name__ == "__main__":