Update app.py

app.py

@@ -155,15 +155,7 @@ def detect_with_model(image: Image.Image, prompt: str, username: str, model_name
     message = f"🔍 Prediction: {prediction} ({confidence}% confidence)\n🧐 Prompt match: {round(prompt_score, 2)}%"
     if prediction == "Real" and model_name.lower() != "real":
         leaderboard_scores[username] = leaderboard_scores.get(username, 0) + score
-        message += "\n🎉 Nice! You fooled the AI. +1 point!"
-
-        image_dir = os.path.join("test", "fake")
-        os.makedirs(image_dir, exist_ok=True)
-        image_id = random.randint(1000000, 9999999)
-        image_filename = f"{image_id}.jpg"
-        image_path = os.path.join(image_dir, image_filename)
-        image.save(image_path)
-            
+        message += "\n🎉 Nice! You fooled the AI. +1 point!"            
     else:
         if model_name.lower() == "real":
             message += "\n You uploaded a real image, this does not count toward the leaderboard!"
@@ -176,6 +168,13 @@ def detect_with_model(image: Image.Image, prompt: str, username: str, model_name
     leaderboard_table = [[name, points] for name, points in sorted_scores]
 
     type_image = "real" if model_name.lower() == "real" else "fake"
+    image_dir = os.path.join("test", type_image)
+    os.makedirs(image_dir, exist_ok=True)
+    image_id = random.randint(10000000, 99999999)
+    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/{type_image}/{image_filename},\"{prompt}\",{type_image},{model_name},test,{prediction.lower()},{username},{datetime.now().isoformat()}\n"