Update app.py

app.py

@@ -5,11 +5,11 @@ import requests
 from serpapi import GoogleSearch
 import tempfile
 
-# Keys
-IMGBB_API_KEY = "YOUR_IMGBB_API_KEY"
-SERPAPI_API_KEY = "YOUR_SERPAPI_API_KEY"
+# Replace with your actual API keys
+IMGBB_API_KEY = "47cf5c6318ae2fc7d1ea3305189942c4"
+SERPAPI_API_KEY = "d70b432deb7b0acfb440fada51e1446974913bb30cc753e78397752336b33db9"
 
-# Load deepfake detection model
+# Load the deepfake detection model
 deepfake_detector = pipeline("image-classification", model="Wvolf/ViT_Deepfake_Detection")
 
 # Upload to ImgBB
@@ -27,7 +27,7 @@ def upload_image_to_imgbb(image, api_key):
     else:
         return None
 
-# Search online
+# Search via SerpAPI
 def search_deepfakes(query, api_key):
     params = {
         "q": query,
@@ -59,21 +59,25 @@ def detect_and_search(image):
         query = f"deepfake {image_url}"
         results = search_deepfakes(query, SERPAPI_API_KEY)
         if results:
-            search_summary = f"\n🔎 Found {len(results)} visually similar images online.\nExample:\n- {results[0]['title']}\n- {results[0]['link']}"
+            first_result = results[0]
+            search_summary = (
+                f"\n🔎 Found {len(results)} visually similar images online.\n"
+                f"Example:\n- {first_result['title']}\n- {first_result['link']}"
+            )
         else:
             search_summary = "\n🔍 No similar deepfakes found online."
     else:
-        search_summary = "\n⚠️ Error uploading image for search."
+        search_summary = "\n⚠️ Error uploading image for reverse search."
 
     return status + search_summary
 
-# Interface
+# Gradio Interface
 iface = gr.Interface(
     fn=detect_and_search,
-    inputs=gr.Image(source="upload", tool="editor", type="pil", label="Upload or Capture Image"),
+    inputs=gr.Image(type="pil", label="Upload or Capture Image"),
     outputs="text",
     title="Deepfake Confidence & Reverse Image Search",
-    description="Upload or capture an image. This tool detects deepfakes and searches the internet to check if similar images already exist.",
+    description="Upload or capture an image. This tool detects deepfakes and checks the internet for visually similar images using SerpAPI."
 )
 
 if __name__ == "__main__":