Update app.py

app.py

@@ -163,38 +163,6 @@ def detect_with_model(image: Image.Image, prompt: str, username: str, model_name
         image_filename = f"{image_id}.jpg"
         image_path = os.path.join(image_dir, image_filename)
         image.save(image_path)
-
-        csv_path = os.path.join("test", "leaderboard_entries.csv")
-        header = "file_name,prompt,label,model,split,user,timestamp\n"
-        csv_line = f"test/fake/{image_filename},\"{prompt}\",fake,{model_name},test,{username},{datetime.now().isoformat()}\n"
-        try:
-            file_exists = os.path.exists(csv_path)
-            with open(csv_path, "a", encoding="utf-8") as f:
-                if not file_exists:
-                    f.write(header)
-                f.write(csv_line)
-
-            from huggingface_hub import HfApi
-
-            api = HfApi()
-            api.upload_file(
-                path_or_fileobj=image_path,
-                path_in_repo=f"test/fake/{image_filename}",
-                repo_id=HUB_REPO_ID,
-                repo_type="dataset",
-                token=HF_TOKEN,
-                commit_message="Add passing image"
-            )
-            api.upload_file(
-                path_or_fileobj=csv_path,
-                path_in_repo="test/leaderboard_entries.csv",
-                repo_id=HUB_REPO_ID,
-                repo_type="dataset",
-                token=HF_TOKEN,
-                commit_message="Update leaderboard CSV"
-            )
-        except Exception as e:
-            print(f"Failed to save image to HF Hub: {e}")
             
     else:
         if model_name.lower() == "real":
@@ -207,6 +175,39 @@ def detect_with_model(image: Image.Image, prompt: str, username: str, model_name
     sorted_scores = sorted(leaderboard_scores.items(), key=lambda x: x[1], reverse=True)
     leaderboard_table = [[name, points] for name, points in sorted_scores]
 
+    type_image = "real" if model_name.lower() == "real" else "fake"
+    csv_path = os.path.join("test", "leaderboard_entries.csv")
+    header = "file_name,prompt,label,model,split,user,timestamp\n"
+    csv_line = f"test/{type_image}/{image_filename},\"{prompt}\",{type_image},{model_name},test,{prediction.lower()},{username},{datetime.now().isoformat()}\n"
+    try:
+        file_exists = os.path.exists(csv_path)
+        with open(csv_path, "a", encoding="utf-8") as f:
+            if not file_exists:
+                f.write(header)
+            f.write(csv_line)
+
+        from huggingface_hub import HfApi
+
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj=image_path,
+            path_in_repo=f"test/{type_image}/{image_filename}",
+            repo_id=HUB_REPO_ID,
+            repo_type="dataset",
+            token=HF_TOKEN,
+            commit_message="Add passing image"
+        )
+        api.upload_file(
+            path_or_fileobj=csv_path,
+            path_in_repo="test/leaderboard_entries.csv",
+            repo_id=HUB_REPO_ID,
+            repo_type="dataset",
+            token=HF_TOKEN,
+            commit_message="Update leaderboard CSV"
+        )
+    except Exception as e:
+        print(f"Failed to save image to HF Hub: {e}")
+
     return (
         message,
         image,