Upload folder using huggingface_hub

Dockerfile

@@ -17,6 +17,7 @@ COPY . .
 ENV PYTHONUNBUFFERED=1
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 ENV GRADIO_SERVER_PORT=7860
+ENV GRADIO_SSR=0
 
 EXPOSE 7860
 

app.py

@@ -29,7 +29,6 @@ def save_persona_base(personas):
     if not HF_TOKEN:
         print("HF_TOKEN not found, skipping upload.")
         return
-    # Use a temporary file to avoid conflicts
     temp_file = "persona_base_upload.json"
     with open(temp_file, 'w', encoding='utf-8') as f:
         json.dump(personas, f, indent=4)
@@ -61,8 +60,6 @@ def render_personas_to_markdown(personas):
         md += f"**Age**: {age} | **Gender**: {gender} | **Nationality**: {nationality}\n\n"
         md += f"**Occupation**: {occupation}\n\n"
         md += f"**Description**: {description}\n\n"
-        
-        # Add a small expandable section for the full JSON
         md += f"<details><summary>View Full JSON</summary>\n\n```json\n{json.dumps(p, indent=2)}\n```\n\n</details>\n\n"
         md += "---\n"
     return md
@@ -88,7 +85,7 @@ def generate_personas(business_description, customer_profile, num_personas, blab
         )
 
         for i in range(num_personas):
-            current_md = f"Generating persona {i+1}/{num_personas}... ⏳\n\n" + render_personas_to_markdown(all_personas_data)
+            current_md = f"#### 🔄 Generating persona {i+1}/{num_personas}... ⏳\n\n" + render_personas_to_markdown(all_personas_data)
             yield all_personas_data, current_md, gr.update(visible=True)
             
             person = factory.generate_person()
@@ -97,7 +94,7 @@ def generate_personas(business_description, customer_profile, num_personas, blab
                 all_personas_data.append(persona_data)
                 yield all_personas_data, render_personas_to_markdown(all_personas_data), gr.update(visible=True)
 
-        yield all_personas_data, render_personas_to_markdown(all_personas_data), gr.update(visible=False)
+        yield all_personas_data, "### ✅ Generation complete! All personas saved to Tresor.\n\n" + render_personas_to_markdown(all_personas_data), gr.update(visible=False)
 
     except GeneratorExit:
         print("Generation cancelled by user.")
@@ -106,9 +103,12 @@ def generate_personas(business_description, customer_profile, num_personas, blab
     finally:
         if all_personas_data:
             print(f"Saving {len(all_personas_data)} personas...")
-            current_base = load_persona_base()
-            current_base.extend(all_personas_data)
-            save_persona_base(current_base)
+            try:
+                current_base = load_persona_base()
+                current_base.extend(all_personas_data)
+                save_persona_base(current_base)
+            except Exception as se:
+                print(f"Error during final save: {se}")
 
         if original_key is None:
             if "BLABLADOR_API_KEY" in os.environ:
@@ -135,7 +135,7 @@ def find_best_persona(criteria):
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🏭 Tiny Persona Factory")
-    gr.Markdown("Generate realistic personas for your business simulation.")
+    gr.Markdown("Generate realistic personas for your business simulation. Results are automatically saved to the Tresor.")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -183,6 +183,7 @@ def health():
 def api_docs():
     return RedirectResponse(url="/docs")
 
+# Try mounting with ssr=False to avoid port conflicts
 try:
     app = gr.mount_gradio_app(app, demo, path="/", ssr=False)
 except Exception: