Update player.py

player.py

@@ -76,48 +76,144 @@ def show_vid(vid_file,list_hid):
     </video>'''
     return label_html
 
-def load_vid_og(channel):
-    out_html=""
-    label_html=""
+##############################################
+
+def build_space(t_space,t_title,t_description=None,t_name=None,t_redirect=None,t_image=None,t_image_url=None,token=None):
+    
+
+
+    
+    if t_description == None or t_description == "":
+        t_description = f'Embedded App'
+    if t_redirect == None or t_redirect =="":
+        t_redirect = t_space
+    if t_name == None or t_name =="":
+        print (t_space)
+        t_name = t_space.split("//",1)[1]
+        print (t_name)
+        t_name = t_name.strip(".hf.space")
+        print (f't_name::{t_name}')
+
+    
+    model_id=t_name
+    #model_id="omnibus/fff"
+    if token==None or token=="":
+        token = token_self
+    else:
+        token = token
+        pass
+    api = HfApi(token=token)
+    repo_name = get_full_repo_name(model_id=model_id, token=token)
+
     try:
-        file_box = [info.path for info in list_files_info(f'{channel}', repo_type="dataset")]
-        ai_tube_box = [info.path for info in list_files_info(f'{source_dataset}', repo_type="dataset")]
-        for file in file_box:
-            if file.startswith("prompt_"):
-                prompt_slug = file.split("prompt_",1)[1].split(".",1)[0]
-                out_html=f'{out_html}<pre>  • {file}</pre>'
-                #print(prompt_slug)
-                for vid_file in ai_tube_box:
-                    try:
-                        vid_slug=vid_file.split("videos/",1)[1].split(".",1)[0]
-                        #print(vid_slug)
-                        try:
-                            if vid_slug == prompt_slug:
-                                out_html=f'{out_html}<pre>    + {vid_file}</pre>'
-                                
-                                if vid_file.endswith(".json"): 
-                                    r = requests.get(f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/raw/main/videos/{vid_slug}.json')
-                                    #r = requests.get(f'https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/blob/main/videos/00b4bcda-7b4a-40f8-9833-e490425a7b91.mp4')
-                                    json_out=r.json()
-                                    label_html=f'{label_html}{json_out["label"]}<br>{json_out["description"]}<br>'
-                                    print (r.json())
-                                if vid_file.endswith(".mp4"):
-                                    #<source src="https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/resolve/main/videos/{vid_file}" type="video/mp4">
-                                    print(vid_file)
-                                    label_html=f'''{label_html}
-                                    <video src="https://huggingface.co/datasets/jbilcke-hf/ai-tube-index/resolve/main/{vid_file}" controls>
-                                    </video>'''
-                                #uid = uuid.uuid4()
-                                #file = open(f'{name}/{f_name}','wb')
-                                #file.write(r.content)
-                        except Exception as e:
-                            print (f'error:: {e}')
-                            pass
-                    except Exception:
-                        pass
+        print(t_space.rsplit('.hf', 1)[-1])
+        if t_space.rsplit('.hf', 1)[-1] == ".space":        
+            repo_url = api.create_repo(
+                repo_id=model_id,
+                repo_type="space",
+                space_sdk="static",
+                private=False,
+            )
+            print(f"""Space Built at {repo_name}""")
+        else:
+            model_id=None
+            repo_name=None
+            t_space=None
+            t_image=None
+            t_image_url=None
+            api=None
+            pass
+            return gr.HTML.update(f"""Invalid Direct URL: must be a Huggingface Space<br>Be sure URL is in lowercase""")  
     except Exception as e:
-        print(e)
-    return(out_html,label_html)
+        return gr.HTML(f"""{str(e)}""")
+    
+    if t_image_url != None and t_image_url != "" and t_image==None:
+        t_image=t_image_url
+        print(f"Using image URL {t_image}")
+        pass
+    elif t_image != None:
+        img1 = Image.open(t_image)
+        img1.thumbnail((500,500), Image.Resampling.LANCZOS)
+        img1.save(f'tmpim-{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"
+        
+########## norm 
+        with open("template/ai.html", "r") as f:
+            app = f.read()
+        app = app.replace("$space", t_space)
+        app = app.replace("$title", t_title)
+        app = app.replace("$description", t_description)
+        app = app.replace("$image", t_image)
+        app = app.replace("$redirect", t_redirect)
+
+        with open("ai.html", "w") as f:
+            f.write(app)
+
+        api.upload_file(
+            path_or_fileobj="ai.html",
+            path_in_repo="ai.html",
+            repo_id=repo_name,
+            token=token,
+            repo_type="space",
+        )
+        os.remove("ai.html")
+
+
+        return gr.HTML(f'''<center>Your Interactive Twitter Card Embed Link is:<br><a href="{t_link}" target="_blank">{t_link}</a><br>
+                                  Your Interactive Mastodon Card Embed Link is:<br><a href="{t_link3}" target="_blank">{t_link3}</a><br>
+                                  Your Interactive Gab Card Embed Link is:<br><a href="{t_link2}" target="_blank">{t_link2}</a><br>
+                                  <center><br>''')
+    except Exception as e:
+        return gr.HTML(f"""{str(e)}""")
+
+    
+#########################################    
+
+
+
 with gr.Blocks() as app:
     with gr.Row():
         channel = gr.Dropdown()