Update app.py

app.py

@@ -9,6 +9,7 @@ uid = uuid.uuid4()
 #token_self = os.environ['HF_TOKEN']
 o=os.environ['P']
 model_list = [
+    "",
     "CompVis/stable-diffusion-v1-4",
     "runwayml/stable-diffusion-v1-5",
     "prompthero/openjourney",
@@ -36,9 +37,10 @@ types=[
     "A",
     "B",
     "C",
+    "R1",
 ]
  
-def build_space(token,t_name,s_type,m_type):
+def build_space(token,t_name,s_type,m_type=None,r_type=None,d_app=None,d_css=None):
     uid = uuid.uuid4()
     model_id=t_name
     if token==None or token=="":
@@ -48,7 +50,6 @@ def build_space(token,t_name,s_type,m_type):
         pass
     api = HfApi(token=token)
     repo_name = get_full_repo_name(model_id=model_id, token=token)
-
     try:
         repo_url = api.create_repo(
             repo_id=model_id,
@@ -66,6 +67,8 @@ def build_space(token,t_name,s_type,m_type):
             with open("a.py", "r") as f:
                 app = f.read()
             app = app.replace("$name2", m_type)
+            app = app.replace("$app2", d_app)
+            app = app.replace("$css2", d_css)
    
             with open(f"{uid}at.py", "w") as f:
                 f.write(app)
@@ -82,7 +85,9 @@ def build_space(token,t_name,s_type,m_type):
             with open("b.py", "r") as f:
                 app = f.read()
             app = app.replace("$name2", m_type)
-   
+            app = app.replace("$app2", d_app)
+            app = app.replace("$css2", d_css)
+            
             with open(f"{uid}bt.py", "w") as f:
                 f.write(app)
     
@@ -98,7 +103,9 @@ def build_space(token,t_name,s_type,m_type):
             with open("c.py", "r") as f:
                 app = f.read()
             app = app.replace("$name2", m_type)
-   
+            app = app.replace("$app2", d_app)
+            app = app.replace("$css2", d_css)
+            
             with open(f"{uid}ct.py", "w") as f:
                 f.write(app)
     
@@ -110,6 +117,24 @@ def build_space(token,t_name,s_type,m_type):
                 repo_type="space",
             )
             os.remove(f"{uid}ct.py")  
+        elif s_type == "R1":
+            with open("r1.py", "r") as f:
+                app = f.read()
+            app = app.replace("$name2", r_type)
+            app = app.replace("$app2", d_app)
+            app = app.replace("$css2", d_css)
+            
+            with open(f"{uid}r1t.py", "w") as f:
+                f.write(app)
+    
+            api.upload_file(
+                path_or_fileobj=f"{uid}r1t.py",
+                path_in_repo="app.py",
+                repo_id=repo_name,
+                token=token,
+                repo_type="space",
+            )
+            os.remove(f"{uid}r1t.py")  
             
             return gr.HTML.update(f'''<center>Link to space <a href="https://huggingface.co/spaces/{repo_name}" target="_blank">https://huggingface.co/spaces/{repo_name}</a><center><br>''')
     except Exception as e:
@@ -130,14 +155,20 @@ with gr.Blocks() as build:
         with gr.Column(scale=3):
             with gr.Box():
                 token = gr.Textbox(label="auth")
-                t_name = gr.Textbox(label="Space Name")     
-                s_type = gr.Dropdown(label="Space Type",choices=[t for t in types])
-                m_type = gr.Dropdown(label="Model", choices=[m for m in model_list])
+                t_name = gr.Textbox(label="Space Name") 
+                with gr.Row():
+                    with gr.Column():
+                        s_type = gr.Dropdown(label="Space Type",choices=[t for t in types])
+                        d_app = gr.Textbox(label="App File", value="app.py")
+                        d_css = gr.Textbox(label="CSS File", value="css.css")
+                    with gr.Column():
+                        m_type = gr.Dropdown(label="Model", choices=[m for m in model_list])
+                        r_type = gr.Textbox(label="repo/name")
                 with gr.Row():
                     inbut = gr.Button()
                 output_html = gr.HTML("""""")
                 files=gr.File()
         gr.Column(scale=1)
     pass_btn.click(checkp,pass_box,[no,yes])
-    inbut.click(build_space,[token,t_name,s_type,m_type],output_html)
+    inbut.click(build_space,[token,t_name,s_type,m_type,r_type,d_app,d_css],output_html)
 build.queue(concurrency_count=10).launch()