Update app.py

app.py

@@ -8,7 +8,7 @@ from pathlib import Path
 from huggingface_hub import (create_repo,get_full_repo_name,upload_file,CommitOperationAdd,HfApi)
 uid = uuid.uuid4()
 
-token_self = os.environ['HF_TOKEN']
+#token_self = os.environ['HF_TOKEN']
 
 
 
@@ -24,6 +24,112 @@ def slide_fn(val):
     img2.save(f'tmpim2-{uid}.png')    
     return img2
 
+def make_files(t_name,t_space,t_title,t_description,t_redirect,t_image=None,t_image_url=None,token=None):
+    model_id=t_name
+    files=[]
+    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)
+        files.append(f"./img-{uid}.png",f"./img2-{uid}.png")
+        pass 
+        
+    else: 
+        print("Default Image")
+        t_image="https://huggingface.co/spaces/portal/bin/resolve/main/ai_demo_card.png"
+        files.append(t_image)
+        pass
+        
+    try:
+        api_url = f'https://huggingface.co/api/spaces/{model_id}'
+        t_iframe = t_space
+        t_link3= f"https://{repo_name.replace('/','-').replace('_','-')}.hf.space/masto.html"
+        t_link2= f"https://{repo_name.replace('/','-').replace('_','-')}.hf.space/gab.html"
+        t_link = f"https://{repo_name.replace('/','-').replace('_','-')}.hf.space/ai.html"
+        t_space = f"https://{repo_name.replace('/','-').replace('_','-')}.hf.space/back.html"
+        t_space2 = f"https://{repo_name.replace('/','-').replace('_','-')}.hf.space/back_gab.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)
+        files.append("ai.html")
+
+######## gab
+        with open("template/ai.html", "r") as f:
+            app = f.read()
+        app = app.replace("$space", t_space2)
+        app = app.replace("$title", t_title)
+        app = app.replace("$description", t_description)
+        app = app.replace("$image", t_image2)
+        app = app.replace("$redirect", t_redirect)
+
+        with open("gab.html", "w") as f:
+            f.write(app)
+        files.append("gab.html")
+
+######## masto
+        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_image2)
+        app = app.replace("$redirect", t_redirect)
+
+        with open("masto.html", "w") as f:
+            f.write(app)
+        files.append("masto.html")
+            
+
+##################
+        with open("template/front.html", "r") as f:
+            app = f.read()
+        app = app.replace("$iframe", t_iframe)
+
+        with open("index.html", "w") as f:
+            f.write(app)
+        files.append("index.html")
+
+######### norm    
+        with open("template/back.html", "r") as f:
+            app = f.read()
+        app = app.replace("$iframe", t_iframe)
+        app = app.replace("$pad", "100%")
+        with open("back.html", "w") as f:
+            f.write(app)
+        files.append("back.html")
+
+####### gab
+        with open("template/back.html", "r") as f:
+            app = f.read()
+        app = app.replace("$iframe", t_iframe)
+        app = app.replace("$pad", "56%")
+        with open("back_gab.html", "w") as f:
+            f.write(app)
+        files.append("back_gab.html")
+        return files
+    
+        
+    except Exception as e:
+        return None
 
 def build_space(t_name,t_space,t_title,t_description,t_redirect,t_image=None,t_image_url=None,token=None):
     model_id=t_name
@@ -234,6 +340,7 @@ def build_space(t_name,t_space,t_title,t_description,t_redirect,t_image=None,t_i
                                   <center><br>''')
     except Exception as e:
         return gr.HTML.update(f"""{str(e)}""")
+
 def d_im():
     output="https://huggingface.co/spaces/portal/bin/resolve/main/ai_demo_card.png"
     return output
@@ -281,19 +388,29 @@ with gr.Blocks() as build_app:
                 t_image_url = gr.Textbox(label="Image URL", placeholder="Optional URL Input")
                 t_image = gr.Image(label="Twitter Card Image", type="filepath",interactive=True)
                 with gr.Row():
-                    c_image=gr.Image(label="Cropped Image", type="filepath",interactive=False)
+                    with gr.Column():
+                        c_image=gr.Image(label="Cropped Image", type="filepath",interactive=False)
+                        def_im = gr.Button("Load Default Image")
                     with gr.Column():
                         c_slider=gr.Slider(0,220,value=0,step=1,label="Adjust Vertical Crop")
                         c_upbtn=gr.Button("Update Crop")
-                with gr.Row():
-                    input_button = gr.Button()
-                    def_im = gr.Button("Load Default Image")
-                output_html = gr.HTML("""""")
+                with gr.Tab("Build"):
+                    with gr.Row():
+                        build_button = gr.Button("Build")
+                        gr.Column()
+                    output_html = gr.HTML("""""")
+                with gr.Tab("Download"):
+                    with gr.Row():
+                        dl_button = gr.Button("Download")
+                        gr.Column()
+                    files_out = gr.Files()
         gr.Column(scale=1)
 
         
     t_image.change(up_c,[t_image,c_slider],c_image)
     c_upbtn.click(slide_fn,c_slider,c_image)
-    input_button.click(build_space,[t_name,t_space,t_title,t_description,t_redirect,t_image,t_image_url,option_token],output_html)
+    build_button.click(build_space,[toggle_build,t_name,t_space,t_title,t_description,t_redirect,t_image,t_image_url,option_token],output_html)
+    dl_button.click(build_space,[toggle_build,t_name,t_space,t_title,t_description,t_redirect,t_image,t_image_url,option_token],files_out)
+
     def_im.click(d_im,None,t_image)
 build_app.queue(concurrency_count=10).launch()