Update app.py

Added real image support

app.py

@@ -121,12 +121,12 @@ def compute_prompt_match(image: Image.Image, prompt: str) -> float:
         return 0.0
 
 # --- Main prediction logic ---
-def detect_with_model(image: Image.Image, prompt: str, username: str):
+def detect_with_model(image: Image.Image, prompt: str, username: str, model: str):
     if not username.strip():
         return "Please enter your name.", None, [], gr.update(visible=True), gr.update(visible=False), username
 
     prompt_score = compute_prompt_match(image, prompt)
-    if prompt_score < PROMPT_MATCH_THRESHOLD:
+    if prompt_score < PROMPT_MATCH_THRESHOLD and model.lower() != "real":
         message = f"⚠️ Prompt match too low ({round(prompt_score, 2)}%). Please generate an image that better matches the prompt."
         return message, None, leaderboard, gr.update(visible=True), gr.update(visible=False), username
 
@@ -144,11 +144,14 @@ def detect_with_model(image: Image.Image, prompt: str, username: str):
     score = 1 if prediction == "Real" else 0
 
     message = f"🔍 Prediction: {prediction} ({confidence}% confidence)\n🧐 Prompt match: {round(prompt_score, 2)}%"
-    if prediction == "Real":
+    if prediction == "Real" and model.lower() != "real":
         leaderboard_scores[username] = leaderboard_scores.get(username, 0) + score
         message += "\n🎉 Nice! You fooled the AI. +1 point!"
     else:
-        message += "\n😅 The AI caught you this time. Try again!"
+        if model.lower() == "real":
+            message += "\n You uploaded a real image, this does not count toward the leaderboard!"
+        else:
+            message += "\n😅 The AI caught you this time. Try again!"
 
     save_leaderboard()
 
@@ -215,7 +218,7 @@ with gr.Blocks(css=".gr-button {font-size: 16px !important}") as demo:
 
     submit_btn.click(
         fn=detect_with_model,
-        inputs=[image_input, prompt_input, username_input],
+        inputs=[image_input, prompt_input, username_input, model_input],
         outputs=[
             prediction_output,
             image_output,