Update player.py

player.py

@@ -99,7 +99,7 @@ def build_space(index_sel, vid_file, t_name=None, token=None):
     print(vid_slug)
     #json_in = f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/blob/main/videos/{vid_slug}.json'
     video_in = f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/resolve/main/videos/{vid_slug}.mp4'
-    #thumb_in = f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/blob/main/videos/{vid_slug}.webp'
+    thumb_in = f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/blob/main/videos/{vid_slug}.webp'
 
     r = requests.get(f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/raw/main/videos/{vid_slug}.json')
     json_out=r.json()
@@ -142,61 +142,11 @@ def build_space(index_sel, vid_file, t_name=None, token=None):
         )
     except Exception as e:
         return gr.HTML(f"""{str(e)}""")
-    
-    if t_image != None:
-
-        urllib.request.urlretrieve(f'{t_image}',"gfg.png") 
-        img1 = Image.open("gfg.png") 
-        img1.save(f'img-{uid}.png')
-
-        #img1 = Image.open(t_image)
-        img1.thumbnail((500,500), Image.Resampling.LANCZOS)
-        img1.save(f'img2-{uid}.png')
-        
-        #output_pro=cv2.imread(f'tmpim-{uid}.png')
-        #output_pro2=cv2.imread(f'tmpim2-{uid}.png')
-
-        #cv2.imwrite(f"./img-{uid}.png",output_pro)
-        #cv2.imwrite(f"./img2-{uid}.png",output_pro2)
-        try: 
-            api.upload_file(
-            path_or_fileobj=f"./img-{uid}.png",
-            path_in_repo="card_im.png",
-            repo_id=repo_name,
-            token=token,
-            repo_type="space",
-            )
-            t_image = f"https://{repo_name.replace('/','-').replace('_','-')}.static.hf.space/card_im.png"
-            os.remove(f"./img-{uid}.png")
-
-            print(f"Image Uploaded to: {t_image}")
-        except Exception as e:
-            return gr.HTML(f"""{str(e)}""")       
-        try: 
-            api.upload_file(
-            path_or_fileobj=f"./img2-{uid}.png",
-            path_in_repo="card_im_crop.png",
-            repo_id=repo_name,
-            token=token,
-            repo_type="space",
-            )
-            t_image2 = f"https://{repo_name.replace('/','-').replace('_','-')}.static.hf.space/card_im_crop.png"
-            os.remove(f"./img2-{uid}.png")
-
-            print(f"Image Uploaded to: {t_image2}")
-        except Exception as e:
-            return gr.HTML(f"""{str(e)}""")   
-            
-        pass 
-    else: 
-        print("Default Image")
-        t_image="https://huggingface.co/spaces/portal/bin/resolve/main/ai_demo_card.png"
-        pass
-        
+           
     try:
         api_url = f'https://huggingface.co/api/spaces/{model_id}'
         #t_iframe = t_space
-        t_link = f"https://{repo_name.replace('/','-').replace('_','-')}.static.hf.space/ai.html"
+        t_link = f"https://{repo_name.replace('/','-').replace('_','-')}.static.hf.space/{f_name}.html"
         
 ########## norm 
         with open("template_front.html", "r") as f:
@@ -204,7 +154,7 @@ def build_space(index_sel, vid_file, t_name=None, token=None):
         app = app.replace("$space", video_in)
         app = app.replace("$title", t_title)
         app = app.replace("$description", t_description)
-        app = app.replace("$image", t_image)
+        app = app.replace("$image", thumb_in)
         app = app.replace("$redirect", t_redirect)
 
         with open("ai.html", "w") as f: