Upload folder using huggingface_hub

app.py

@@ -43,7 +43,7 @@ def extract_persona_parameters(business_description: str, customer_profile: str)
     - `custom_life_attitude` (str): The life attitude of the persona.
     - `life_story` (str): A brief life story of the persona.
     - `interests_hobbies` (str): Interests and hobbies of the persona.
-    - `attribute_count` (float): Attribute richness, default to 200.
+    - `attribute_count` (float): Attribute richness, default to 350.
 
     You must return a valid JSON object containing exactly these keys.
     """
@@ -76,7 +76,7 @@ def extract_persona_parameters(business_description: str, customer_profile: str)
                 if key not in extracted_json:
                     # provide defaults for missing ones
                     if key in ['age', 'attribute_count']:
-                        extracted_json[key] = 200 if key == 'attribute_count' else 30
+                        extracted_json[key] = 350 if key == 'attribute_count' else 30
                     else:
                         extracted_json[key] = "Unknown"
                         
@@ -96,7 +96,7 @@ def extract_persona_parameters(business_description: str, customer_profile: str)
         "custom_life_attitude": "Optimistic",
         "life_story": "A standard professional background with a passion for their field.",
         "interests_hobbies": "Technology, Reading",
-        "attribute_count": 200
+        "attribute_count": 350
     }
 
 def generate_personas(business_description, customer_profile, num_personas, blablador_api_key=None):
@@ -148,7 +148,7 @@ def generate_personas(business_description, customer_profile, num_personas, blab
                 custom_life_attitude=str(params.get("custom_life_attitude", "Optimistic")),
                 life_story=str(params.get("life_story", "A standard professional background with a passion for their field.")),
                 interests_hobbies=str(params.get("interests_hobbies", "Technology, Reading")),
-                attribute_count=float(params.get("attribute_count", 200)),
+                attribute_count=float(params.get("attribute_count", 350)),
                 api_name="/generate_persona"
             )
             

test_api.py

@@ -25,7 +25,7 @@ def test_extract_persona_parameters_success():
     with patch('tinytroupe.openai_utils.client') as mock_client:
         mock_instance = MagicMock()
         mock_instance.send_message.return_value = {
-            "content": '{"age": 25, "gender": "Female", "occupation": "Engineer", "city": "NYC", "country": "USA", "custom_values": "Innovation", "custom_life_attitude": "Positive", "life_story": "A story", "interests_hobbies": "Coding", "attribute_count": 200}'
+            "content": '{"age": 25, "gender": "Female", "occupation": "Engineer", "city": "NYC", "country": "USA", "custom_values": "Innovation", "custom_life_attitude": "Positive", "life_story": "A story", "interests_hobbies": "Coding", "attribute_count": 350}'
         }
         mock_client.return_value = mock_instance
         
@@ -45,7 +45,7 @@ def test_generate_personas(mock_client_class):
             "age": 25, "gender": "Female", "occupation": "Engineer", 
             "city": "NYC", "country": "USA", "custom_values": "Innovation", 
             "custom_life_attitude": "Positive", "life_story": "A story", 
-            "interests_hobbies": "Coding", "attribute_count": 200
+            "interests_hobbies": "Coding", "attribute_count": 350
         }
         
         # We need an API key to pass the check