Update app.py

app.py

@@ -6,8 +6,8 @@ from google_img_source_search import ReverseImageSearcher
 from PIL import Image
 import os 
 import uuid
-uid=uuid.uuid4()
-size_js="""
+uid = str(uuid.uuid4())
+size_js = """
     function imgSize(){
         var myImg = document.getElementsByClassName("my_im");
         var realWidth = myImg.naturalWidth;
@@ -17,34 +17,27 @@ size_js="""
 
 def dl(inp):
     out = None
-    out_file=[]
+    out_file = []
     try:
-        inp_out=inp.replace("https://","")
-        inp_out=inp_out.replace("/","_").replace(".","_").replace("=","_").replace("?","_")
-        print(f"Downloading video from URL: {inp}")
+        inp_out = inp.replace("https://", "")
+        inp_out = inp_out.replace("/", "_").replace(".", "_").replace("=", "_").replace("?", "_")
         if "twitter" in inp:
             os.system(f'yt-dlp "{inp}" --extractor-arg "twitter:api=syndication" --trim-filenames 160 -o "{uid}/{inp_out}.mp4" -S res,mp4 --recode mp4')  
         else:
             os.system(f'yt-dlp "{inp}" --trim-filenames 160 -o "{uid}/{inp_out}.mp4" -S res,mp4 --recode mp4')          
         
         out = f"{uid}/{inp_out}.mp4"
-        print(f"Download complete: {out}")
+        print(out)
     except Exception as e:
-        print(f"Error in downloading video: {e}")
-    return out,gr.HTML(""),"",""
+        print(e)
+    return out, gr.HTML(""), "", ""
 
-def process_vid(file,cur_frame,every_n):
+def process_vid(file, cur_frame, every_n):
     new_video_in = str(file)
-    print(f"Processing video file: {new_video_in}")
     capture = cv2.VideoCapture(new_video_in)    
-    if not capture.isOpened():
-        print(f"Error: Could not open video file {new_video_in}")
-        return (gr.HTML(f'Error: Could not open video file {new_video_in}'), "", "")
-    
     frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
-    print(f"Total frames in video: {frame_count}")
     rev_img_searcher = ReverseImageSearcher()
-    html_out=""
+    html_out = ""
     count = int(every_n)
     if cur_frame == "" or cur_frame is None:
         start_frame = 0
@@ -54,22 +47,18 @@ def process_vid(file,cur_frame,every_n):
         for i in range(start_frame, frame_count-1):
             if count == int(every_n):
                 count = 1
-                print(f"Processing frame: {i}")
                 capture.set(cv2.CAP_PROP_POS_FRAMES, i)
                 ret, frame_f = capture.read()
                 if not ret:
-                    print(f"Frame {i} could not be captured.")
+                    print(f"Failed to capture frame at {i}")
                     continue
                 cv2.imwrite(f"{uid}-vid_tmp{i}.png", frame_f)
                 out = os.path.abspath(f"{uid}-vid_tmp{i}.png")
-                print(f"Saved frame {i} as image: {out}")
                 out_url = f'https://nymbo-reverse-image.hf.space/file={out}'
                 res = rev_img_searcher.search(out_url)
-                print(f"Reverse image search results for frame {i}: {res}")
                 out_cnt = 0
                 if len(res) > 0:
                     for search_item in res:
-                        print(f'Found result {out_cnt + 1} for frame {i}')
                         out_cnt += 1
                         html_out = f"""{html_out}
                         <div>
@@ -80,17 +69,12 @@ def process_vid(file,cur_frame,every_n):
                         </div>"""
                     return (gr.HTML(f'<h1>Total Found: {out_cnt}</h1><br>{html_out}'), f"Found frame: {i}", i+int(every_n))
             count += 1
-            print(f"Skipping frame {i + 1}")
     except Exception as e:
-        print(f"Error during video processing: {e}")
         return (gr.HTML(f'{e}'), "", "")
-    print("No matching frames found in video.")
     return (gr.HTML('No frame matches found.'), "", "")
 
-def process_im(file,url):
-    print(f"Processing image with URL: {url}")
+def process_im(file, url):
     if not url.startswith("https://nymbo"):
-        print(f"URL does not match required pattern: {url}")
         return url
     else:
         read_file = Image.open(file)
@@ -98,28 +82,26 @@ def process_im(file,url):
         action_input = f"{uid}-tmp.png"
         out = os.path.abspath(action_input)
         out_url = f'https://nymbo-reverse-image.hf.space/file={out}'
-        print(f"Processed image saved as: {out}")
         return (out_url)
 
 def rev_im(image):
-    print(f"Reversing image search for: {image}")
+    out_list = []
+    out_im = []
+    html_out = """"""
     image = cv2.imread(image)
-    if image is None:
-        print(f"Error: Could not read image {image}")
-        return gr.HTML(f'Error: Could not read image {image}')
-    
     cv2.imwrite(f"{uid}-im_tmp.png", image)
     out = os.path.abspath(f"{uid}-im_tmp.png")
-    print(f"Image saved for reverse search: {out}")
     out_url = f'https://nymbo-reverse-image.hf.space/file={out}'    
     rev_img_searcher = ReverseImageSearcher()
     res = rev_img_searcher.search(out_url)
-    print(f"Reverse image search results: {res}")
     count = 0
-    html_out = """"""
     for search_item in res:
         count += 1
-        print(f"Found result {count} for image")
+        out_dict = {
+            'Title': f'{search_item.page_title}',
+            'Site': f'{search_item.page_url}',
+            'Img': f'{search_item.image_url}',
+        }
         html_out = f"""{html_out}
         <div>
         Title: {search_item.page_title}<br>
@@ -130,13 +112,11 @@ def rev_im(image):
         
     return (gr.HTML(f'<h1>Total Found: {count}</h1><br>{html_out}'))
 
-
 with gr.Blocks() as app:
     with gr.Row():
         gr.Column()
         with gr.Column():
-            
-            source_tog = gr.Radio(choices=["Image", "Video"], value="Image")
+            source_tog = gr.Radio(choices=["Image","Video"], value="Image")
             with gr.Box(visible=True) as im_box:
                 inp_url = gr.Textbox(label="Image URL")
                 load_im_btn = gr.Button("Load Image")
@@ -152,24 +132,18 @@ with gr.Blocks() as app:
                 with gr.Row():
                     go_btn_vid = gr.Button("Start")
                     next_btn = gr.Button("Next")
-                
-        gr.Column()
+        gr.Column()            
     with gr.Row():
         html_out = gr.HTML("""""")
     with gr.Row(visible=False):
         hid_box = gr.Textbox()
-    
     def shuf(tog):
-        print(f"Toggling source type to: {tog}")
         if tog == "Image":
             return gr.update(visible=True), gr.update(visible=False)
         if tog == "Video":
             return gr.update(visible=False), gr.update(visible=True)    
-    
     def load_image(url):
-        print(f"Loading image from URL: {url}")
         return url
-    
     im_load = load_im_btn.click(load_image, inp_url, inp_im)
     next_btn.click(process_vid, [inp_vid, hid_box, every_n], [html_out, stat_box, hid_box])
     vid_load = vid_url_btn.click(dl, vid_url, [inp_vid, html_out, stat_box, hid_box])