Upload folder using huggingface_hub

app.py

@@ -42,6 +42,109 @@ simulation_manager = DummySimulationManager()
 
 REMOTE_BACKEND = "https://auxteam-tiny-factory.hf.space"
 
+
+# =========================================================================
+# External/Hybrid Persona Generation Logic
+# =========================================================================
+
+CONFIG = {
+    "persona_generation": {
+        "mode": "hybrid",  # internal, external, hybrid
+        "external_api_url": "THzva/deeppersona-experience",
+        "fallback_to_internal": True
+    },
+    "action_generation": {
+        "use_internal_llm": True,
+        "external_llm_endpoint": None
+    }
+}
+
+class InternalPersonaGenerator:
+    def generate(self, business_description, customer_profile, num_personas, api_key=None):
+        if api_key:
+            os.environ["OPENAI_API_KEY"] = api_key
+        get_tinytroupe_modules()
+        from tinytroupe.factory.tiny_person_factory import TinyPersonFactory
+        factory = TinyPersonFactory(context=f"{business_description} {customer_profile}")
+        personas = factory.generate_people(number_of_people=int(num_personas))
+        return [p._persona for p in personas]
+
+class ExternalPersonaGenerator:
+    def generate(self, business_description, customer_profile, num_personas, api_key=None):
+        return generate_personas(business_description, customer_profile, num_personas, api_key)
+
+class HybridPersonaGenerator:
+    def __init__(self, use_external=True):
+        self.use_external = use_external
+        
+    def generate(self, business_description, customer_profile, num_personas, api_key=None):
+        # We can try external first, and if it fails or returns less, fallback
+        personas = ExternalPersonaGenerator().generate(business_description, customer_profile, num_personas, api_key)
+        if not personas and CONFIG["persona_generation"]["fallback_to_internal"]:
+            return InternalPersonaGenerator().generate(business_description, customer_profile, num_personas, api_key)
+        return personas
+
+def create_persona(name: str, business_description: str, customer_profile: str, num_personas: int, api_key=None):
+    mode = CONFIG["persona_generation"]["mode"]
+    if mode == "external":
+        return ExternalPersonaGenerator().generate(business_description, customer_profile, num_personas, api_key)
+    elif mode == "hybrid":
+        try:
+            return HybridPersonaGenerator(use_external=True).generate(business_description, customer_profile, num_personas, api_key)
+        except Exception as e:
+            print("Hybrid generation exception:", e)
+            if CONFIG["persona_generation"]["fallback_to_internal"]:
+                return InternalPersonaGenerator().generate(business_description, customer_profile, num_personas, api_key)
+            raise
+    else:
+        return InternalPersonaGenerator().generate(business_description, customer_profile, num_personas, api_key)
+
+def get_APIEnhancedTinyPerson():
+    get_tinytroupe_modules()
+    class APIEnhancedTinyPerson(TinyPerson):
+        """TinyPerson with external API integration"""
+        def __init__(self, name: str, use_api: bool = False, **api_config):
+            super().__init__(name)
+            self.use_api = use_api
+            from gradio_client import Client
+            self.api_client = Client(api_config.get('api_url', CONFIG["persona_generation"]["external_api_url"])) if use_api else None
+            
+        def enhance_with_api(self, **params):
+            if not self.use_api:
+                raise RuntimeError("API mode not enabled")
+            
+            result = self.api_client.predict(
+                age=float(params.get('age', self._get_attr('age', 25))),
+                gender=str(params.get('gender', self._get_attr('gender', 'Female'))),
+                occupation=str(params.get('occupation', self._get_attr('occupation', 'Professional'))),
+                city=str(params.get('city', self._get_attr('residence', 'Unknown'))),
+                country=str(params.get('country', self._get_attr('nationality', 'Unknown'))),
+                custom_values=str(params.get('custom_values', self._get_attr('beliefs', 'Hardworking'))),
+                custom_life_attitude=str(params.get('custom_life_attitude', self._get_attr('personality', 'Positive'))),
+                life_story=str(params.get('life_story', self._get_attr('other_facts', 'Standard story'))),
+                interests_hobbies=str(params.get('interests_hobbies', self._get_attr('preferences', 'Reading'))),
+                attribute_count=float(params.get('attribute_count', 350.0)),
+                api_name="/generate_persona"
+            )
+            
+            # The result from DeepPersona is raw string. Let's merge it into the profile text.
+            self._merge_persona_data({"full_profile_text": result})
+            return self
+            
+        def _get_attr(self, key, default):
+            return getattr(self, '_persona', {}).get(key, default)
+            
+        def _merge_persona_data(self, api_data: dict):
+            for key, value in api_data.items():
+                if hasattr(self, f'_{key}'):
+                    setattr(self, f'_{key}', value)
+                else:
+                    self._persona[key] = value
+                    
+    return APIEnhancedTinyPerson
+
+# =========================================================================
+
 def generate_personas(business_description, customer_profile, num_personas, api_key=None):
     if api_key:
         os.environ["OPENAI_API_KEY"] = api_key
@@ -306,8 +409,8 @@ def update_persona_preview(file):
 
 def generate_personas_router(business_description, customer_profile, num_personas, method, example_file, api_key=None):
     if method == "DeepPersona":
-        # Call the existing DeepPersona implementation
-        return generate_personas(business_description, customer_profile, num_personas, api_key)
+        CONFIG["persona_generation"]["mode"] = "hybrid"
+        return create_persona("Dynamic", business_description, customer_profile, num_personas, api_key)
     elif method == "TinyTroupe":
         if api_key:
             os.environ["OPENAI_API_KEY"] = api_key