Upload folder using huggingface_hub

app.py

@@ -169,7 +169,10 @@ def handle_generate(theme, customer_profile, num_personas):
     except Exception as e:
         yield f"Error during generation: {str(e)}", None, None
 
-def start_and_monitor_sessions(repo_name, branch_name, personas, tasks, url):
+def start_and_monitor_sessions(personas, tasks, url):
+    repo_name = "JsonLord/tiny_web"
+    branch_name = "main"
+
     if not JULES_API_KEY:
         yield "Error: JULES_API_KEY not set.", ""
         return
@@ -277,10 +280,6 @@ with gr.Blocks() as demo:
             num_personas_input = gr.Number(label="Number of Personas", value=1, precision=0)
             url_input = gr.Textbox(label="Target URL", value="https://example.com")
             
-            repo_selector = gr.Dropdown(label="GitHub Repository", choices=get_user_repos(), value="JsonLord/tiny_web")
-            branch_selector = gr.Dropdown(label="Branch", choices=["main"], value="main")
-            
-            repo_selector.change(fn=get_repo_branches, inputs=repo_selector, outputs=branch_selector)
             
             generate_btn = gr.Button("Generate Personas & Tasks")
             
@@ -304,7 +303,7 @@ with gr.Blocks() as demo:
     
     start_session_btn.click(
         fn=start_and_monitor_sessions,
-        inputs=[repo_selector, branch_selector, persona_display, task_list_display, url_input],
+        inputs=[persona_display, task_list_display, url_input],
         outputs=[status_output, report_output]
     )