added token and modified /submit

app.py

@@ -905,7 +905,7 @@ fastapi_app.add_middleware(
 )
 
 _bearer_scheme = HTTPBearer(auto_error=False)
-_api_bearer_token = os.getenv("API_BEARER_TOKEN", "changeme")
+_api_bearer_token = os.getenv("API_BEARER_TOKEN", "logicgo_123")
 
 def _verify_bearer_token(credentials: Optional[HTTPAuthorizationCredentials] = Depends(_bearer_scheme)):
     if not _bearer_scheme:
@@ -990,6 +990,22 @@ async def submit(
             progress=progress_stub,
         )
 
+        # --- Filter only final outputs ---
+        final_images = [
+            os.path.basename(p)
+            for p in (output_files or [])
+            if not any(tag in os.path.basename(p) for tag in ["cropped_faces", "restored_faces", "cmp"])
+        ]
+        final_zips = [
+            os.path.basename(p)
+            for p in (zip_files or [])
+            if "restored_images" in os.path.basename(p)
+        ]
+
+        def _dl_url(name: str) -> str:
+            return f"/download/{name}"
+
+        # Save to Mongo if available
         if _mongo_collection is not None:
             try:
                 _mongo_collection.insert_one({
@@ -1005,29 +1021,28 @@ async def submit(
                         "with_model_name": with_model_name,
                     },
                     "outputs": {
-                        "images": [os.path.basename(p) for p in (output_files or [])],
-                        "zips": [os.path.basename(p) for p in (zip_files or [])],
+                        "images": final_images,
+                        "zips": final_zips,
                     },
                 })
             except Exception:
                 pass
 
-        def _dl_url(name: str) -> str:
-            return f"/download/{name}"
-
         return {
-            "output_images": [os.path.basename(p) for p in (output_files or [])],
-            "output_zips": [os.path.basename(p) for p in (zip_files or [])],
+            "output_images": final_images,
+            "output_zips": final_zips,
             "download_urls": {
-                "images": [_dl_url(os.path.basename(p)) for p in (output_files or [])],
-                "zips": [_dl_url(os.path.basename(p)) for p in (zip_files or [])],
+                "images": [_dl_url(p) for p in final_images],
+                "zips": [_dl_url(p) for p in final_zips],
             },
         }
+
     except HTTPException:
         raise
     except Exception as e:
         raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e))
 
+
 @fastapi_app.get("/download/{filename}")
 def download(filename: str, _: bool = Depends(_verify_bearer_token)):
     safe_name = os.path.basename(filename)