Update app.py

app.py

@@ -12,45 +12,82 @@ from Gradio_UI import GradioUI
 
     
 @tool
-def transform_belief_and_motivate(negative_belief: str) -> str:
-    """A tool that transforms a negative belief into a positive one and provides a motivational phrase.
+    def belief_transformer(negative_belief: str, intensity_level: int) -> str:
+    """A tool that transforms negative beliefs into positive ones and generates motivational quotes
+    
     Args:
-        negative_belief: A negative belief or thought (e.g., "Я не справлюсь с этой задачей").
+        negative_belief: The negative belief that needs to be transformed
+        intensity_level: Level of emotional intensity (1-5) to adjust response tone
+    
     Returns:
-        A string containing a positive belief and a motivational phrase.
+        str: A formatted string containing the transformed belief and a motivational quote
     """
-    if not negative_belief.strip():
-        return "Sorry: No belief provided - Nothing to transform 😐"
-
-    # Простая база позитивных убеждений для типичных ситуаций
-    positive_beliefs = [
-        "Я справлюсь со всем, что встретится на моем пути.",
-        "Каждый день я становлюсь сильнее и увереннее.",
-        "Я способен(на) на большее, чем думаю.",
-        "Все трудности временные, я нахожу решения.",
-        "Я достоин(на) успеха и добьюсь своих целей.",
-        "Ошибки — это опыт, который делает меня мудрее."
-    ]
-
-    # Простая база мотивирующих фраз
-    motivational_phrases = [
-        "Ты сильнее, чем кажется. Вперёд!",
-        "Каждый шаг приближает тебя к мечте.",
-        "Не сдавайся — успех уже рядом!",
-        "Ты можешь всё. Просто начни!",
-        "Лучший момент действовать — сейчас!",
-        "Дорогу осилит идущий. Иди вперёд!"
-    ]
-
-    positive_belief = random.choice(positive_beliefs)
-    motivational_phrase = random.choice(motivational_phrases)
-
-    return f"Позитивное убеждение: {positive_belief}\nМотивирующая фраза: {motivational_phrase}"
-
-
-result = transform_belief_and_motivate("Я не смогу достичь своей цели")
-print(result)
+    import re
+    from typing import Dict, List
+    
+    # Define transformation patterns
+    belief_patterns: Dict[str, str] = {
+        r"я не могу": "я учусь и развиваюсь, чтобы",
+        r"я недостаточно": "я становлюсь более",
+        r"я никогда не": "я на пути к тому, чтобы",
+        r"это слишком сложно": "это вызов, который помогает мне расти",
+        r"я боюсь": "я смело встречаю",
+        r"я плохой": "я совершенствуюсь в",
+        r"у меня не получится": "я делаю шаги к успеху в",
+    }
+    
+    # Motivational quotes based on intensity
+    quotes: Dict[int, List[str]] = {
+        1: [
+            "Каждый шаг - это движение вперёд.",
+            "Прогресс начинается с принятия себя.",
+            "Малые изменения приводят к большим результатам."
+        ],
+        2: [
+            "Ты сильнее, чем думаешь.",
+            "В тебе есть всё необходимое для успеха.",
+            "Каждая трудность - это возможность для роста."
+        ],
+        3: [
+            "Твоё стремление к улучшению уже делает тебя особенным!",
+            "Вера в себя - твой главный союзник.",
+            "Ты создаёшь свою реальность своими мыслями."
+        ],
+        4: [
+            "Твой потенциал безграничен! Действуй!",
+            "Ты способен на невероятные достижения!",
+            "Каждый день - новая возможность стать лучше!"
+        ],
+        5: [
+            "ТВОЯ СИЛА ДУХА НЕПОБЕДИМА!",
+            "ТЫ СОЗДАН ДЛЯ ВЕЛИКИХ СВЕРШЕНИЙ!",
+            "ТВОЁ ВРЕМЯ ПРИШЛО! ДЕЙСТВУЙ СЕЙЧАС!"
+        ]
+    }
+    
+    def transform_belief(text: str) -> str:
+        transformed = text.lower()
+        for pattern, replacement in belief_patterns.items():
+            transformed = re.sub(pattern, replacement, transformed)
+        return transformed.capitalize()
+    
+    def get_quote(level: int) -> str:
+        import random
+        level = max(1, min(5, level))  # Ensure level is between 1 and 5
+        return random.choice(quotes[level])
+    
+    # Transform the belief and get a quote
+    positive_belief = transform_belief(negative_belief)
+    motivation = get_quote(intensity_level)
+    
+    # Format the response
+    response = f"""Трансформация убеждения:
+Было: {negative_belief}
+Стало: {positive_belief}
 
+Мотивация: {motivation}"""
+    
+    return response
 
 
 final_answer = FinalAnswerTool()