Update app.py

app.py

@@ -55,21 +55,22 @@ def show_f(repo,name,token):
 def load_space(r_name,s_name,d_app):
     print(f'https://huggingface.co/spaces/{r_name}/{s_name}/raw/main/{d_app}')
     r = requests.get(f'https://huggingface.co/spaces/{r_name}/{s_name}/raw/main/{d_app}') 
-    sf=d_app.split(".",1)[1]
-    pf=d_app.split(".",1)[0]
-    f_name=f'{pf}.{sf}'
-    file = open(f'{uid}/{f_name}','w')
+    #sf=d_app.split(".",1)[1]
+    #pf=d_app.split(".",1)[0]
+    #f_name=f'{pf}.{sf}'
+    
+    file = open(f'{uid}/{d_app}','w')
     file.writelines(r.text)
     file.close()
     return r.text, f_name
-def repl(rs,rt,uu):
-    file = open(f'{uid}/{uu}','r')
+def repl(rs,rt,d_app):
+    file = open(f'{uid}/{d_app}','r')
     r_file = file.read()
     r_file = r_file.replace(f'{rs}',f'{rt}')
-    o_file = open(f'{uid}/mod_{uu}','w')
+    o_file = open(f'{uid}/mod_{d_app}','w')
     o_file = o_file.write(r_file)
     file.close()
-    return r_file, f'{uid}/mod_{uu}'
+    return r_file, f'{uid}/mod_{d_app}'
 
 def get_f(r_name,spaces,token):
     api = HfApi(token=token)
@@ -98,19 +99,25 @@ def get_f(r_name,spaces,token):
             
     return(files_b,spaces_b)
 
-def build_space(token,r_name,s_name,d_app,uu):
+def build_space(token,r_name,s_name,d_app,rd_name,sd_name,dd_app):
     if token==None or token=="":
         token = token_self
     else:
         token = token
         pass
+    if dd_app=="":
+        dd_app=d_app
+    if rd_name=="":
+        rd_name=r_name
+    if sd_name=="":
+        sd_name=s_name
     api = HfApi(token=token)
-    repo_name=f'{r_name}/{s_name}'
+    repo_name=f'{rd_name}/{sd_name}'
     print (repo_name)
     try:
         api.upload_file(
-            path_or_fileobj=f"{uid}/mod_{uu}",
-            path_in_repo=f"{d_app}",
+            path_or_fileobj=f"{uid}/mod_{d_app}",
+            path_in_repo=f"{dd_app}",
             repo_id=repo_name,
             token=token,
             repo_type="space",
@@ -136,12 +143,14 @@ with gr.Blocks() as build:
             with gr.Column(scale=3):
                 with gr.Box():
                     with gr.Row():
-                        r_name = gr.Textbox(label="Repo") 
-                        token = gr.Textbox(label="auth")
-                       
-                    with gr.Row():
-                        s_name = gr.Dropdown(label="Spaces", choices=[])
-                        d_app = gr.Dropdown(label="Files", choices=[])
+                        with gr.Column():
+                            r_name = gr.Textbox(label="Source Repo") 
+                            s_name = gr.Dropdown(label="Spaces", choices=[])
+                            d_app = gr.Dropdown(label="Files", choices=[])                       
+                        with gr.Column():
+                            rd_name = gr.Textbox(label="Dest Repo") 
+                            sd_name = gr.Dropdown(label="Spaces", choices=[])
+                            dd_app = gr.Dropdown(label="Files", choices=[])              
                     with gr.Row():
                         s_btn = gr.Button("Show")
                         l_btn = gr.Button("Load")
@@ -150,7 +159,9 @@ with gr.Blocks() as build:
                     r_w = gr.Textbox(label="with:")
                     r_btn = gr.Button("Replace")
                 with gr.Column():
-                    inbut = gr.Button("Update Repo")
+                    with gr.Row():
+                        token = gr.Textbox(label="auth")
+                        inbut = gr.Button("Update Repo")
                     output_html = gr.HTML("""""")
             gr.Column(scale=1)
         with gr.Accordion("Files",open=False):
@@ -183,5 +194,5 @@ with gr.Blocks() as build:
     r_btn.click(repl,[r_s,r_w,uu], [m_t,files])
     
     pass_btn.click(checkp,pass_box,[no,yes])
-    inbut.click(build_space,[token,r_name,s_name,d_app,uu],output_html)
+    inbut.click(build_space,[token,r_name,s_name,d_app,dd_app],output_html)
 build.queue(concurrency_count=10).launch()