Update app.py

app.py

@@ -8,6 +8,10 @@ import os
 import uuid
 import subprocess
 import html
+import imgbbpy  # Adding imgbb to upload images for accessible URLs
+
+# Initialize imgbb client with your API key (you need to provide your key)
+imgbb_client = imgbbpy.SyncClient("YOUR_IMGBB_API_KEY")
 
 # Function to download video from a given URL using yt-dlp
 def dl(inp):
@@ -63,7 +67,10 @@ def process_vid(file, cur_frame, every_n):
                 # Save the current frame as an image file
                 frame_path = f"{uid}-vid_tmp{i}.png"
                 cv2.imwrite(frame_path, frame_f)
-                out_url = f'https://omnibus-reverse-image.hf.space/file={frame_path}'
+                
+                # Upload the frame to imgbb to get a publicly accessible URL
+                response = imgbb_client.upload(file=frame_path)
+                out_url = response.url  # Get the public URL from imgbb response
                 
                 # Perform reverse image search on the extracted frame
                 res = rev_img_searcher.search(out_url)
@@ -98,7 +105,10 @@ def process_im(file, url):
         read_file.save(f"{uid}-tmp.png")
         action_input = f"{uid}-tmp.png"
         out = os.path.abspath(action_input)
-        out_url = f'https://omnibus-reverse-image.hf.space/file={out}'
+        
+        # Upload the image to imgbb to get a publicly accessible URL
+        response = imgbb_client.upload(file=out)
+        out_url = response.url  # Get the public URL from imgbb response
         return out_url
 
 # Function to perform reverse image search on a given image
@@ -108,9 +118,12 @@ def rev_im(image):
     
     # Read the image using OpenCV and save it to a temporary file
     image = cv2.imread(image)
-    cv2.imwrite(f"{uid}-im_tmp.png", image)
-    out = os.path.abspath(f"{uid}-im_tmp.png")
-    out_url = f'https://omnibus-reverse-image.hf.space/file={out}'    
+    tmp_image_path = f"{uid}-im_tmp.png"
+    cv2.imwrite(tmp_image_path, image)
+    
+    # Upload the image to imgbb to get a publicly accessible URL
+    response = imgbb_client.upload(file=tmp_image_path)
+    out_url = response.url  # Get the public URL from imgbb response
     
     # Perform reverse image search
     rev_img_searcher = ReverseImageSearcher()