Upload folder using huggingface_hub

analysis_template.md

@@ -1,7 +1,3 @@
-# CONTEXT FILE: contexts/context_{{report_id}}.md
-# COMMAND: Use `read_file` or equivalent to read the context file at the path above.
-# It contains the persona JSON and the tasks list for this analysis.
-
 # UX STRATEGIST TEMPLATE (CLIENT-GRADE)
 
 You are a Senior UX Strategist and Researcher.
@@ -17,6 +13,8 @@ You will simulate a persona performing tasks, but the final output must read lik
 
 ## 1. Variables
 
+- Persona: {{persona_context}}
+- Tasks: {{tasks_list}}
 - Target URL: {{url}}
 - Report ID: {{report_id}}
 
@@ -26,7 +24,7 @@ You will simulate a persona performing tasks, but the final output must read lik
 
 ### Task Execution & Data Collection
 
-1. **Sequential Execution**: Perform the 10 tasks provided in the external context file one by one.
+1. **Sequential Execution**: Perform the 10 tasks provided in `{{tasks_list}}` one by one.
 2. **Coordinate Tracking**: For every click or interaction, record the (x, y) coordinates relative to the viewport.
 3. **Heatmap Generation**: After completing all tasks, generate "Average User Journey Heatmaps" by overlaying the recorded interaction points onto screenshots of the relevant pages.
    - Save these heatmap images as PNG files in `/user_experience_reports/heatmaps/`.

app.py

@@ -55,8 +55,9 @@ def patch_tinytroupe():
         if "_raw_model_call_parallel" not in content:
             content = content.replace("class OpenAIClient:", "class OpenAIClient:" + parallel_helper)
 
-        # 2. Ensure alias-large is used (revert any previous fast patches)
-        content = content.replace('"alias-fast"', '"alias-large"')
+        # 2. Ensure alias-huge is used (alias-large is deprecated/down)
+        content = content.replace('"alias-fast"', '"alias-huge"')
+        content = content.replace('"alias-large"', '"alias-huge"')
 
         # 3. Handle 502 errors by waiting 35 seconds and setting a parallel retry flag
         # We need to modify the send_message loop
@@ -68,8 +69,8 @@ def patch_tinytroupe():
         if 'if parallel_retry:' not in content:
             old_call = "response = self._raw_model_call(model, chat_api_params)"
             new_call = """if parallel_retry:
-                        logger.info("Attempting parallel call to alias-large and alias-huge.")
-                        response = self._raw_model_call_parallel(["alias-large", "alias-huge"], chat_api_params)
+                        logger.info("Attempting parallel call to alias-huge and alias-fast.")
+                        response = self._raw_model_call_parallel(["alias-huge", "alias-fast"], chat_api_params)
                         if isinstance(response, Exception):
                             raise response
                     else:
@@ -310,35 +311,6 @@ def upload_persona_to_pool(persona_data):
     except Exception as e:
         print(f"Error uploading persona to pool: {e}")
 
-def upload_context_to_github(repo_name, session_id, persona, tasks):
-    if not gh:
-        add_log("ERROR: GitHub client not initialized for context upload.")
-        return None
-    
-    file_path = f"contexts/context_{session_id}.md"
-    content = f"""# Analysis Context for Session {session_id}
-
-## Persona
-{json.dumps(persona, indent=2)}
-
-## Tasks
-{json.dumps(tasks, indent=2)}
-"""
-    try:
-        repo = gh.get_repo(repo_name)
-        try:
-            # Check if exists (unlikely for new session)
-            existing = repo.get_contents(file_path, ref="main")
-            repo.update_file(file_path, f"Update context for {session_id}", content, existing.sha, branch="main")
-        except:
-            repo.create_file(file_path, f"Add context for {session_id}", content, branch="main")
-        
-        add_log(f"Successfully uploaded context file to {repo_name} main branch.")
-        return f"https://github.com/{repo_name}/blob/main/{file_path}"
-    except Exception as e:
-        add_log(f"ERROR uploading context to GitHub: {e}")
-        return None
-
 def select_or_create_personas(theme, customer_profile, num_personas, force_method=None, example_file=None):
     if force_method == "Example Persona" and example_file:
         add_log(f"Loading example persona from {example_file}...")
@@ -424,7 +396,7 @@ def select_or_create_personas(theme, customer_profile, num_personas, force_metho
 
     try:
         response = client.chat.completions.create(
-            model="alias-large",
+            model="alias-huge",
             messages=[{"role": "user", "content": prompt}]
         )
         content = response.choices[0].message.content
@@ -463,7 +435,7 @@ def generate_persona_from_deeppersona(theme, customer_profile):
     if not client:
         return None
 
-    # Step 1: Breakdown profile into parameters using LLM alias-large
+    # Step 1: Breakdown profile into parameters using LLM alias-huge
     prompt = f"""
     You are an expert in persona creation. 
     Break down the following business theme and customer profile into detailed attributes for a persona.
@@ -487,7 +459,7 @@ def generate_persona_from_deeppersona(theme, customer_profile):
 
     try:
         response = client.chat.completions.create(
-            model="alias-large",
+            model="alias-huge",
             messages=[{"role": "user", "content": prompt}],
             response_format={"type": "json_object"}
         )
@@ -564,9 +536,9 @@ def generate_personas(theme, customer_profile, num_personas):
     
     add_log("Falling back to TinyTroupe logic for remaining personas...")
     
-    # Ensure alias-large is used
-    config_manager.update("model", "alias-large")
-    config_manager.update("reasoning_model", "alias-large")
+    # Ensure alias-huge is used
+    config_manager.update("model", "alias-huge")
+    config_manager.update("reasoning_model", "alias-huge")
 
     context = f"A company related to {theme}. Target customers: {customer_profile}"
 
@@ -644,7 +616,7 @@ def generate_tasks(theme, customer_profile, url):
     Do not include any other text in your response.
     """
 
-    models_to_try = ["alias-large", "alias-huge", "alias-fast"]
+    models_to_try = ["alias-huge", "alias-fast", "alias-large"]
 
     for attempt in range(5):
         try:
@@ -656,7 +628,7 @@ def generate_tasks(theme, customer_profile, url):
                 response = call_llm_parallel(client, models_to_try, [{"role": "user", "content": prompt}], response_format={"type": "json_object"})
             else:
                 response = client.chat.completions.create(
-                    model="alias-large",
+                    model="alias-huge",
                     messages=[{"role": "user", "content": prompt}],
                     response_format={"type": "json_object"}
                 )
@@ -695,7 +667,7 @@ def handle_generate(theme, customer_profile, num_personas, method, example_file,
             if ex_personas:
                 current_profile = ex_personas[0].get('minibio', customer_profile)
 
-        yield "Generating tasks...", None, None, None
+        yield "Thinking...", None, None, None
         tasks = generate_tasks(theme, current_profile, url)
         tasks_text = "\n".join(tasks) if isinstance(tasks, list) else str(tasks)
 
@@ -742,21 +714,15 @@ def start_and_monitor_sessions(personas, tasks, url, session_id):
     sessions = []
     jules_uuids = []
     
-    # Upload context file to main branch
-    context_url = upload_context_to_github(repo_name, session_id, personas[0], tasks)
-    if not context_url:
-        add_log("Warning: Failed to upload context file. Jules might not find it.")
-    else:
-        add_log(f"Waiting 5 seconds for GitHub propagation...")
-        time.sleep(5)
-
     for persona in personas:
         # Use provided session_id or append to it if multiple personas?
         # For simplicity, we use session_id as the report_id too
         report_id = session_id
         
         # Format prompt
-        prompt = template.replace("{{url}}", url)
+        prompt = template.replace("{{persona_context}}", json.dumps(persona))
+        prompt = prompt.replace("{{tasks_list}}", json.dumps(tasks))
+        prompt = prompt.replace("{{url}}", url)
         prompt = prompt.replace("{{report_id}}", report_id)
         prompt = prompt.replace("{{blablador_api_key}}", BLABLADOR_API_KEY if BLABLADOR_API_KEY else "YOUR_API_KEY")
 
@@ -770,7 +736,7 @@ def start_and_monitor_sessions(personas, tasks, url, session_id):
             "sourceContext": {
                 "source": f"sources/github/{repo_name}",
                 "githubRepoContext": {
-                    "startingBranch": session_id
+                    "startingBranch": "main"
                 }
             },
             "automationMode": "AUTO_CREATE_PR",

verify_thinking.py

@@ -0,0 +1,45 @@
+import asyncio
+from playwright.async_api import async_playwright
+
+async def verify_thinking_status():
+    async with async_playwright() as p:
+        browser = await p.chromium.launch(headless=True)
+        page = await browser.new_page()
+        try:
+            await page.goto("http://localhost:7860", timeout=60000)
+            print("Page loaded.")
+            
+            # Fill mandatory fields
+            await page.get_by_label("Theme").fill("Test Theme")
+            await page.get_by_label("Target URL").fill("https://example.com")
+            
+            # Click Generate
+            await page.get_by_role("button", name="Generate Personas & Tasks").click()
+            print("Clicked Generate button.")
+            
+            # Check for Thinking... status
+            # It might be in the status textbox
+            status_box = page.get_by_label("Status")
+            
+            # Use a loop to check for the text since it's transient
+            found = False
+            for _ in range(20):
+                val = await status_box.input_value()
+                if "Thinking..." in val:
+                    print(f"Found status: {val}")
+                    found = True
+                    break
+                await asyncio.sleep(0.5)
+            
+            if not found:
+                print("Thinking... status not found or appeared too quickly.")
+
+            await page.screenshot(path="thinking_verification.png")
+            
+        except Exception as e:
+            print(f"Error: {e}")
+        finally:
+            await browser.close()
+
+if __name__ == "__main__":
+    asyncio.run(verify_thinking_status())