Update app.py

app.py

@@ -1,480 +1,415 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+import gradio as gr
 import datetime
-import requests
-import pytz
-import yaml
 import json
 import re
-import tempfile
 import os
 from typing import Dict, List, Any
-from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI
+import openai
 
-@tool
-def speech_emotion_analyzer(text: str, context: str = "therapy") -> str:
-    """
-    Analyzes patient's emotional state from speech text.
-    
-    Args:
-        text: Patient's speech text
-        context: Context (therapy, assessment, crisis)
-    
-    Returns:
-        Analysis of emotional state and recommendations
-    """
-    if not text or len(text.strip()) < 10:
-        return "Error: Text must be at least 10 characters long for analysis."
-    
-    # Emotion indicators for analysis
-    emotion_indicators = {
-        "depression": {
-            "keywords": ["sad", "hopeless", "tired", "meaningless", "empty", "lonely", 
-                        "depressed", "despair", "miserable", "worthless", "numb"],
-            "patterns": [r"can't.*sleep", r"no.*energy", r"everything.*bad", r"don't.*want", 
-                        r"life.*no.*meaning", r"better.*dead"]
-        },
-        "anxiety": {
-            "keywords": ["anxious", "worried", "nervous", "scared", "panic", "tense",
-                        "anxiety", "worry", "fear", "stress", "phobia"],
-            "patterns": [r"what.*if", r"afraid.*that", r"can't.*cope", r"heart.*racing",
-                        r"hands.*shaking", r"can't.*breathe", r"panic.*attack"]
-        },
-        "anger": {
-            "keywords": ["angry", "furious", "irritated", "annoyed", "hate", "mad",
-                        "rage", "aggression", "hatred", "frustration"],
-            "patterns": [r"fed up", r"sick of", r"can't.*stand", r"want.*hit",
-                        r"everything.*irritates", r"pissed off"]
-        },
-        "stress": {
-            "keywords": ["stress", "pressure", "overwhelmed", "exhausted", "burned out",
-                        "burnout", "fatigue", "drained"],
-            "patterns": [r"too.*much.*work", r"can't.*keep up", r"headache", r"no.*time",
-                        r"swamped", r"working.*nonstop"]
-        },
-        "trauma": {
-            "keywords": ["trauma", "flashbacks", "nightmares", "memories", "avoiding",
-                        "ptsd", "triggered", "shock"],
-            "patterns": [r"keep.*remembering", r"bad.*dreams", r"can't.*forget",
-                        r"avoid.*places", r"triggers"]
-        }
-    }
-    
-    # Analyze text
-    text_lower = text.lower()
-    detected_emotions = {}
-    
-    for emotion, indicators in emotion_indicators.items():
-        score = 0
-        matches = []
-        
-        # Check keywords
-        for keyword in indicators["keywords"]:
-            if keyword in text_lower:
-                score += 1
-                matches.append(keyword)
-        
-        # Check patterns
-        for pattern in indicators["patterns"]:
-            if re.search(pattern, text_lower):
-                score += 2
-                matches.append(f"pattern: {pattern}")
-        
-        if score > 0:
-            detected_emotions[emotion] = {"score": score, "matches": matches}
-    
-    # Check for crisis indicators
-    crisis_indicators = ["suicide", "kill myself", "end it all", "not worth living", "better off dead"]
-    crisis_risk = any(indicator in text_lower for indicator in crisis_indicators)
-    
-    # Check positive indicators
-    positive_indicators = ["happy", "joy", "good", "great", "wonderful", "excited", "grateful", "love"]
-    positive_count = sum(1 for indicator in positive_indicators if indicator in text_lower)
-    
-    # Create report
-    report = "## 🧠 Emotional State Analysis\n\n"
-    
-    if detected_emotions:
-        report += "### 📊 Detected Emotional States:\n"
-        for emotion, data in sorted(detected_emotions.items(), key=lambda x: x[1]["score"], reverse=True):
-            emotion_names = {
-                "depression": "😔 Depression signs",
-                "anxiety": "😰 Anxiety", 
-                "anger": "😠 Anger/Irritation",
-                "stress": "😵 Stress",
-                "trauma": "💔 Trauma-related"
-            }
-            intensity = "High" if data['score'] >= 4 else "Medium" if data['score'] >= 2 else "Low"
-            report += f"- **{emotion_names[emotion]}** (intensity: {intensity})\n"
-            report += f"  Found indicators: {', '.join(data['matches'][:3])}\n\n"
-    else:
-        report += "✅ No clear signs of emotional distress detected.\n\n"
-    
-    # Positive aspects
-    if positive_count > 0:
-        report += f"😊 **Positive indicators**: Found {positive_count} positive marker(s)\n\n"
-    
-    # Crisis assessment
-    if crisis_risk:
-        report += "🚨 **CRITICAL: Suicide risk indicators detected!**\n"
-        report += "📞 Immediate consultation and safety assessment recommended.\n"
-        report += "Emergency services: 911, Crisis hotline: 988\n\n"
-    
-    # Treatment recommendations
-    report += "### 🎯 Treatment Recommendations:\n"
-    
-    if "depression" in detected_emotions:
-        report += "- 🧘 **CBT (Cognitive Behavioral Therapy)** for negative thought patterns\n"
-        report += "- 💊 Consider psychiatric consultation for medication support\n"
-        report += "- 🏃 Behavioral activation (increase pleasant activities)\n"
-    
-    if "anxiety" in detected_emotions:
-        report += "- 🫁 **Relaxation techniques**: breathing exercises, progressive muscle relaxation\n"
-        report += "- 🎭 **Exposure therapy** for phobic avoidance\n"
-        report += "- 🧠 **Mindfulness therapy** for anxious thoughts\n"
-    
-    if "anger" in detected_emotions:
-        report += "- 🎯 **Anger management training** and self-regulation techniques\n"
-        report += "- 🔍 **Trigger identification** and behavior patterns\n"
-        report += "- 🤝 **Communication skills training**\n"
-    
-    if "stress" in detected_emotions:
-        report += "- ⏰ **Stress management** and time management techniques\n"
-        report += "- ⚖️ **Work-life balance** assessment and adjustment\n"
-        report += "- 🛡️ **Coping strategies** for stressful situations\n"
-    
-    if "trauma" in detected_emotions:
-        report += "- 👁️ **EMDR therapy** for traumatic memories\n"
-        report += "- 🏠 **Trauma-informed approach** in therapy\n"
-        report += "- 🧘 **Somatic techniques** for body-based trauma symptoms\n"
-    
-    if not detected_emotions or positive_count > 2:
-        report += "- 💪 **Supportive therapy** to maintain resources\n"
-        report += "- 🎯 **Personal growth** and life goals development\n"
-    
-    report += f"\n📅 *Analysis date: {datetime.datetime.now().strftime('%m/%d/%Y %H:%M')}*"
-    report += f"\n🌍 *Context: {context}*"
-    
-    return report
-
-@tool
-def research_therapy_techniques(disorder: str, age_group: str = "adult") -> str:
-    """
-    Searches for modern therapy methods for specific disorders via DuckDuckGo.
-    
-    Args:
-        disorder: Mental disorder or problem
-        age_group: Age group (child, teen, adult, elderly)
-    
-    Returns:
-        Information about modern therapy methods
-    """
-    search_tool = DuckDuckGoSearchTool()
-    
-    # Create search queries
-    search_queries = [
-        f"evidence-based therapy {disorder} {age_group} 2024",
-        f"latest treatment {disorder} psychology",
-        f"CBT DBT therapy {disorder} research"
-    ]
-    
-    results = []
-    for query in search_queries:
-        try:
-            search_result = search_tool(query)
-            results.append(search_result)
-        except Exception as e:
-            results.append(f"Search error for '{query}': {str(e)}")
-    
-    # Create report
-    report = f"# 🔬 Therapy Methods Research\n\n"
-    report += f"**Disorder/Problem**: {disorder}\n"
-    report += f"**Age group**: {age_group}\n"
-    report += f"**Search date**: {datetime.datetime.now().strftime('%m/%d/%Y %H:%M')}\n\n"
-    
-    report += "## 📚 Search Results:\n\n"
-    
-    for i, result in enumerate(results, 1):
-        report += f"### Search {i}:\n"
-        report += f"{result}\n\n"
-        report += "---\n\n"
-    
-    report += "## 💡 Recommendations:\n"
-    report += "- Verify the currency of found information\n"
-    report += "- Cross-reference with clinical guidelines\n"
-    report += "- Consider individual client characteristics\n"
-    report += "- Consult with colleagues when needed\n"
-    
-    return report
-
-@tool
-def create_therapy_worksheet(therapy_type: str, client_age: int, issue: str) -> str:
-    """
-    Creates a therapeutic worksheet/exercise for client.
-    
-    Args:
-        therapy_type: Type of therapy (cbt, dbt, art, mindfulness)
-        client_age: Client's age
-        issue: Main problem
-    
-    Returns:
-        Structured therapeutic worksheet
-    """
-    if client_age < 5 or client_age > 100:
-        return "Error: Age must be between 5 and 100."
-    
-    therapy_type = therapy_type.lower()
-    valid_types = ["cbt", "dbt", "art", "mindfulness"]
-    
-    if therapy_type not in valid_types:
-        return f"Error: Therapy type must be one of: {', '.join(valid_types)}"
-    
-    # Determine level by age
-    if client_age < 12:
-        level = "child"
-    elif client_age < 18:
-        level = "teen" 
-    else:
-        level = "adult"
-    
-    worksheets = {
-        "cbt": {
-            "child": {
-                "title": "🌈 My Thoughts and Feelings",
-                "exercises": [
-                    "🌤️ **Mood Diary**: Draw or mark your mood with an emoji each day",
-                    "🤔 **Thought Catcher**: Write down thoughts when sad or scared",
-                    "🎯 **Helper Thoughts**: Find 3 good thoughts for each bad thought",
-                    "⭐ **My Successes**: Write one good thing you did each day"
-                ]
-            },
-            "teen": {
-                "title": "🧠 Cognitive Restructuring",
-                "exercises": [
-                    "📊 **ABC Model**: Situation (A) → Thought (B) → Feeling/Behavior (C)",
-                    "🔍 **Thought Detective**: Find evidence FOR and AGAINST your beliefs",
-                    "⚖️ **Alternative Thoughts**: Find 2-3 realistic alternatives for negative thoughts",
-                    "📝 **Thought Diary**: Daily record of automatic thoughts"
-                ]
-            },
-            "adult": {
-                "title": "🔄 Cognitive Behavioral Therapy",
-                "exercises": [
-                    "🔄 **Thought-Feeling-Behavior Cycle**: Analyze connections in situations",
-                    "⚡ **Cognitive Distortions**: Identify and correct thinking errors",
-                    "🎯 **Behavioral Experiments**: Test catastrophic beliefs",
-                    "📈 **Graded Tasks**: Step-by-step overcoming avoidance"
-                ]
-            }
-        },
-        "dbt": {
-            "teen": {
-                "title": "🌊 Emotion Regulation Skills",
-                "exercises": [
-                    "🌊 **TIPP**: Temperature, Intense exercise, Paced breathing, Paired muscle relaxation",
-                    "🚦 **STOP**: Stop → Take a breath → Observe → Proceed mindfully",
-                    "🎭 **Emotion Diary**: Track emotion intensity (1-10) with triggers",
-                    "🧘 **Mindfulness**: 5-10 minutes daily breathing meditation"
-                ]
-            },
-            "adult": {
-                "title": "⚖️ Dialectical Behavior Therapy",
-                "exercises": [
-                    "⚖️ **Wise Mind**: Balance emotional and rational mind",
-                    "🛡️ **Crisis Skills**: TIPP, distraction, self-soothing",
-                    "🤝 **DEAR MAN**: Technique for healthy communication",
-                    "🎯 **Radical Acceptance**: Practice accepting unchangeable reality"
-                ]
-            }
-        },
-        "art": {
-            "child": {
-                "title": "🎨 Art Therapy Exercises",
-                "exercises": [
-                    "🎨 **Feeling Colors**: Draw your feelings using different colors",
-                    "🏠 **House of Emotions**: Draw a house with different feelings in each room",
-                    "🦋 **Feeling Mask**: Make and color a mask showing outside and inside feelings",
-                    "🌈 **Color Your Day**: Color different parts of your day by emotions"
-                ]
+class PsychologyToolsApp:
+    def __init__(self):
+        self.name = "Psychology Tools Suite"
+        self.description = "AI-powered tools for mental health professionals"
+        
+    def speech_emotion_analyzer(self, text: str, context: str = "therapy") -> str:
+        """Analyzes patient's emotional state from speech text."""
+        if not text or len(text.strip()) < 10:
+            return "Error: Text must be at least 10 characters long for analysis."
+        
+        # Emotion indicators for analysis
+        emotion_indicators = {
+            "depression": {
+                "keywords": ["sad", "hopeless", "tired", "meaningless", "empty", "lonely", 
+                            "depressed", "despair", "miserable", "worthless", "numb"],
+                "patterns": [r"can't.*sleep", r"no.*energy", r"everything.*bad", r"don't.*want", 
+                            r"life.*no.*meaning", r"better.*dead"]
             },
-            "teen": {
-                "title": "🎭 Creative Expression",
-                "exercises": [
-                    "🖼️ **Identity Collage**: Create a visual collage about 'Who am I?'",
-                    "🗺️ **Emotion Map**: Draw a map of your feelings",
-                    "📖 **Visual Journal**: Keep a journal in drawings instead of words",
-                    "🎭 **Metaphor Images**: Draw your problem as a symbol"
-                ]
+            "anxiety": {
+                "keywords": ["anxious", "worried", "nervous", "scared", "panic", "tense",
+                            "anxiety", "worry", "fear", "stress", "phobia"],
+                "patterns": [r"what.*if", r"afraid.*that", r"can't.*cope", r"heart.*racing",
+                            r"hands.*shaking", r"can't.*breathe", r"panic.*attack"]
             },
-            "adult": {
-                "title": "🖌️ Art Therapy for Adults",
-                "exercises": [
-                    "🌳 **Life Tree**: Roots = values, trunk = strength, crown = goals",
-                    "🚪 **Doors of Possibility**: Draw life choices and paths",
-                    "⚡ **Energy Portrait**: Express inner state through abstract art",
-                    "🔄 **Transformation**: Before and After visualization of changes"
-                ]
+            "anger": {
+                "keywords": ["angry", "furious", "irritated", "annoyed", "hate", "mad",
+                            "rage", "aggression", "hatred", "frustration"],
+                "patterns": [r"fed up", r"sick of", r"can't.*stand", r"want.*hit",
+                            r"everything.*irritates", r"pissed off"]
             }
-        },
-        "mindfulness": {
-            "child": {
-                "title": "🐸 Mindfulness for Kids",
-                "exercises": [
-                    "🐸 **Frog Pose**: Sit still like a frog and listen to sounds",
-                    "🌬️ **Belly Breathing**: Watch a toy rise and fall on your belly",
-                    "👀 **5-4-3-2-1**: Find 5 things you see, 4 you hear, 3 you can touch...",
-                    "🍯 **Slow Eating**: Eat one raisin very slowly and mindfully"
-                ]
+        }
+        
+        # Analyze text
+        text_lower = text.lower()
+        detected_emotions = {}
+        
+        for emotion, indicators in emotion_indicators.items():
+            score = 0
+            matches = []
+            
+            for keyword in indicators["keywords"]:
+                if keyword in text_lower:
+                    score += 1
+                    matches.append(keyword)
+            
+            for pattern in indicators["patterns"]:
+                if re.search(pattern, text_lower):
+                    score += 2
+                    matches.append(f"pattern: {pattern}")
+            
+            if score > 0:
+                detected_emotions[emotion] = {"score": score, "matches": matches}
+        
+        # Check for crisis indicators
+        crisis_indicators = ["suicide", "kill myself", "end it all", "not worth living"]
+        crisis_risk = any(indicator in text_lower for indicator in crisis_indicators)
+        
+        # Create report
+        report = "## 🧠 Emotional State Analysis\n\n"
+        
+        if detected_emotions:
+            report += "### 📊 Detected Emotional States:\n"
+            for emotion, data in sorted(detected_emotions.items(), key=lambda x: x[1]["score"], reverse=True):
+                emotion_names = {
+                    "depression": "😔 Depression signs",
+                    "anxiety": "😰 Anxiety", 
+                    "anger": "😠 Anger/Irritation"
+                }
+                intensity = "High" if data['score'] >= 4 else "Medium" if data['score'] >= 2 else "Low"
+                report += f"- **{emotion_names.get(emotion, emotion)}** (intensity: {intensity})\n"
+        else:
+            report += "✅ No clear signs of emotional distress detected.\n\n"
+        
+        if crisis_risk:
+            report += "\n🚨 **CRITICAL: Suicide risk indicators detected!**\n"
+            report += "📞 Immediate consultation recommended. Crisis hotline: 988\n\n"
+        
+        report += f"\n📅 *Analysis date: {datetime.datetime.now().strftime('%m/%d/%Y %H:%M')}*"
+        
+        return report
+    
+    def create_therapy_worksheet(self, therapy_type: str, client_age: int, issue: str) -> str:
+        """Creates a therapeutic worksheet/exercise for client."""
+        if client_age < 5 or client_age > 100:
+            return "Error: Age must be between 5 and 100."
+        
+        therapy_type = therapy_type.lower()
+        valid_types = ["cbt", "dbt", "mindfulness"]
+        
+        if therapy_type not in valid_types:
+            return f"Error: Therapy type must be one of: {', '.join(valid_types)}"
+        
+        # Determine level by age
+        if client_age < 12:
+            level = "child"
+        elif client_age < 18:
+            level = "teen" 
+        else:
+            level = "adult"
+        
+        worksheets = {
+            "cbt": {
+                "child": {
+                    "title": "🌈 My Thoughts and Feelings",
+                    "exercises": [
+                        "🌤️ **Mood Diary**: Draw or mark your mood with an emoji each day",
+                        "🤔 **Thought Catcher**: Write down thoughts when sad or scared",
+                        "🎯 **Helper Thoughts**: Find 3 good thoughts for each bad thought"
+                    ]
+                },
+                "teen": {
+                    "title": "🧠 Cognitive Restructuring",
+                    "exercises": [
+                        "📊 **ABC Model**: Situation → Thought → Feeling/Behavior",
+                        "🔍 **Thought Detective**: Find evidence FOR and AGAINST beliefs",
+                        "📝 **Thought Diary**: Daily record of automatic thoughts"
+                    ]
+                },
+                "adult": {
+                    "title": "🔄 Cognitive Behavioral Therapy",
+                    "exercises": [
+                        "🔄 **Thought-Feeling-Behavior Cycle**: Analyze connections",
+                        "⚡ **Cognitive Distortions**: Identify thinking errors",
+                        "🎯 **Behavioral Experiments**: Test catastrophic beliefs"
+                    ]
+                }
             },
-            "teen": {
-                "title": "🧘 Teen Mindfulness",
-                "exercises": [
-                    "📱 **Digital Detox**: 30-60 minutes without devices",
-                    "🚶 **Mindful Walking**: Walk slowly noticing each step",
-                    "💭 **Thought Watching**: Watch thoughts like clouds passing",
-                    "❤️ **Body Scan**: Relax and notice each body part"
-                ]
+            "dbt": {
+                "teen": {
+                    "title": "🌊 Emotion Regulation Skills",
+                    "exercises": [
+                        "🌊 **TIPP**: Temperature, Intense exercise, Paced breathing",
+                        "🚦 **STOP**: Stop → Take a breath → Observe → Proceed",
+                        "🎭 **Emotion Diary**: Track emotion intensity (1-10)"
+                    ]
+                },
+                "adult": {
+                    "title": "⚖️ Dialectical Behavior Therapy",
+                    "exercises": [
+                        "⚖️ **Wise Mind**: Balance emotional and rational mind",
+                        "🛡️ **Crisis Skills**: TIPP, distraction, self-soothing",
+                        "🤝 **DEAR MAN**: Healthy communication technique"
+                    ]
+                }
             },
-            "adult": {
-                "title": "🧘‍♀️ Mindfulness Practice",
-                "exercises": [
-                    "🧘 **Formal Meditation**: 20-30 minutes daily sitting practice",
-                    "🍽️ **Mindful Eating**: Full attention to taste, smell, texture",
-                    "💼 **Work Mindfulness**: Regular pauses to return to present",
-                    "😴 **Sleep Meditation**: Body scan and breathing for better sleep"
-                ]
+            "mindfulness": {
+                "child": {
+                    "title": "🐸 Mindfulness for Kids",
+                    "exercises": [
+                        "🐸 **Frog Pose**: Sit still and listen to sounds",
+                        "🌬️ **Belly Breathing**: Watch toy rise and fall on belly",
+                        "👀 **5-4-3-2-1**: Find 5 things you see, 4 you hear..."
+                    ]
+                },
+                "teen": {
+                    "title": "🧘 Teen Mindfulness",
+                    "exercises": [
+                        "📱 **Digital Detox**: 30-60 minutes without devices",
+                        "🚶 **Mindful Walking**: Notice each step",
+                        "💭 **Thought Watching**: Observe thoughts like clouds"
+                    ]
+                },
+                "adult": {
+                    "title": "🧘‍♀️ Mindfulness Practice",
+                    "exercises": [
+                        "🧘 **Formal Meditation**: 20-30 minutes daily practice",
+                        "🍽️ **Mindful Eating**: Full attention to food",
+                        "💼 **Work Mindfulness**: Regular present-moment pauses"
+                    ]
+                }
             }
         }
-    }
-    
-    # Get appropriate exercises
-    therapy_data = worksheets.get(therapy_type, {})
-    level_data = therapy_data.get(level, therapy_data.get("adult", {}))
-    
-    if not level_data:
-        return f"Exercises for {therapy_type} therapy and age {client_age} not yet available."
-    
-    # Create worksheet
-    worksheet = f"# {level_data['title']}\n\n"
-    worksheet += f"**👤 Client**: {client_age} years old\n"
-    worksheet += f"**🎯 Main issue**: {issue}\n"
-    worksheet += f"**🔬 Therapy type**: {therapy_type.upper()}\n"
-    worksheet += f"**📅 Date created**: {datetime.datetime.now().strftime('%m/%d/%Y')}\n\n"
-    
-    worksheet += "## 📋 Exercises:\n\n"
-    
-    for i, exercise in enumerate(level_data['exercises'], 1):
-        worksheet += f"### {i}. {exercise}\n\n"
-        
-        # Add space for notes
-        worksheet += "**📝 Notes:**\n"
-        worksheet += "```\n"
-        worksheet += "Date completed: ___________\n"
-        worksheet += "Observations: \n\n\n"
-        worksheet += "Emotions during exercise: \n\n"
-        worksheet += "Insights: \n\n"
-        worksheet += "```\n\n"
-    
-    worksheet += "---\n\n"
-    worksheet += "## 📖 Instructions:\n"
-    worksheet += "✅ Complete exercises at your own pace\n"
-    worksheet += "✅ Record all observations - this is important for progress\n"
-    worksheet += "✅ Discuss results with your therapist\n"
-    worksheet += "✅ Stop if you feel too uncomfortable and seek support\n\n"
-    
-    worksheet += "## 🆘 When to get help:\n"
-    worksheet += "⚠️ If exercises cause severe anxiety or panic\n"
-    worksheet += "⚠️ If you have thoughts of self-harm\n"
-    worksheet += "⚠️ If symptoms worsen\n\n"
+        
+        # Get appropriate exercises
+        therapy_data = worksheets.get(therapy_type, {})
+        level_data = therapy_data.get(level, therapy_data.get("adult", {}))
+        
+        if not level_data:
+            return f"Exercises for {therapy_type} therapy and age {client_age} not available."
+        
+        # Create worksheet
+        worksheet = f"# {level_data['title']}\n\n"
+        worksheet += f"**👤 Client**: {client_age} years old\n"
+        worksheet += f"**🎯 Main issue**: {issue}\n"
+        worksheet += f"**🔬 Therapy type**: {therapy_type.upper()}\n"
+        worksheet += f"**📅 Date**: {datetime.datetime.now().strftime('%m/%d/%Y')}\n\n"
+        
+        worksheet += "## 📋 Exercises:\n\n"
+        
+        for i, exercise in enumerate(level_data['exercises'], 1):
+            worksheet += f"### {i}. {exercise}\n\n"
+            worksheet += "**Notes:**\n```\nDate: _____\nObservations:\n\n\n```\n\n"
+        
+        worksheet += "---\n"
+        worksheet += "## 📞 Emergency contacts:\n"
+        worksheet += "- Crisis hotline: 988\n"
+        worksheet += "- Emergency: 911\n\n"
+        
+        return worksheet
     
-    worksheet += "📞 **Emergency contacts:**\n"
-    worksheet += "- Crisis hotline: 988\n"
-    worksheet += "- Emergency: 911\n"
-    worksheet += "- Your therapist: ________________\n\n"
+    def mood_tracker_analyzer(self, mood_data: str) -> str:
+        """Analyzes mood tracking data to identify patterns."""
+        try:
+            data = json.loads(mood_data)
+        except:
+            return "Error: Please provide mood data in JSON format.\nExample: [{\"date\": \"2024-01-01\", \"mood\": 7}]"
+        
+        if not isinstance(data, list) or len(data) == 0:
+            return "Error: Mood data should be a list of entries with 'date' and 'mood' fields."
+        
+        # Analyze mood patterns
+        moods = []
+        dates = []
+        
+        for entry in data:
+            if 'mood' in entry and 'date' in entry:
+                moods.append(entry['mood'])
+                dates.append(entry['date'])
+        
+        if not moods:
+            return "Error: No valid mood entries found."
+        
+        # Calculate statistics
+        avg_mood = sum(moods) / len(moods)
+        mood_trend = "stable"
+        
+        if len(moods) > 3:
+            recent_avg = sum(moods[-3:]) / 3
+            older_avg = sum(moods[:-3]) / (len(moods) - 3)
+            
+            if recent_avg > older_avg + 0.5:
+                mood_trend = "improving"
+            elif recent_avg < older_avg - 0.5:
+                mood_trend = "declining"
+        
+        # Create report
+        report = "# 📊 Mood Tracking Analysis\n\n"
+        report += f"**Period**: {dates[0]} to {dates[-1]}\n"
+        report += f"**Entries analyzed**: {len(moods)}\n"
+        report += f"**Average mood**: {avg_mood:.1f}/10\n"
+        report += f"**Trend**: {mood_trend}\n\n"
+        
+        # Check for low mood periods
+        low_mood_days = sum(1 for m in moods if m < 5)
+        if low_mood_days > len(moods) * 0.3:
+            report += "⚠️ **Concern**: Low mood in >30% of entries\n\n"
+        
+        report += "## 💡 Recommendations:\n"
+        
+        if mood_trend == "declining":
+            report += "- 🚨 Schedule mental health appointment\n"
+            report += "- 🧘 Increase self-care activities\n"
+        elif mood_trend == "improving":
+            report += "- ✅ Continue current strategies\n"
+            report += "- 📝 Document what's working\n"
+        else:
+            report += "- 📊 Continue tracking\n"
+            report += "- 🔍 Look for patterns\n"
+        
+        return report
     
-    return worksheet
+    def transcribe_audio_with_whisper(self, audio_file, api_key: str = None) -> str:
+        """Transcribes audio file using OpenAI Whisper API."""
+        try:
+            # Set API key
+            if api_key:
+                openai.api_key = api_key
+            elif os.getenv("OPENAI_API_KEY"):
+                openai.api_key = os.getenv("OPENAI_API_KEY")
+            else:
+                return "Error: Please provide OpenAI API key."
+            
+            # Transcribe audio
+            response = openai.Audio.transcribe(
+                model="whisper-1",
+                file=audio_file
+            )
+            
+            text = response.get("text", "")
+            
+            # Create report
+            report = f"## 🎤 Audio Transcription\n\n"
+            report += f"**Timestamp**: {datetime.datetime.now().strftime('%m/%d/%Y %H:%M')}\n\n"
+            report += f"### 📝 Transcribed Text:\n\n{text}\n\n"
+            
+            # Analyze emotions
+            emotion_analysis = self.speech_emotion_analyzer(text)
+            report += "---\n\n" + emotion_analysis
+            
+            return report
+            
+        except Exception as e:
+            return f"Error during transcription: {str(e)}"
 
-@tool
-def mood_tracker_analyzer(mood_data: str) -> str:
-    """
-    Analyzes mood tracking data to identify patterns.
-    
-    Args:
-        mood_data: JSON string with mood entries
-    
-    Returns:
-        Analysis report with patterns and recommendations
-    """
-    try:
-        data = json.loads(mood_data)
-    except:
-        return "Error: Please provide mood data in JSON format."
-    
-    if not isinstance(data, list) or len(data) == 0:
-        return "Error: Mood data should be a non-empty list of entries."
-    
-    # Analyze mood patterns
-    moods = []
-    dates = []
-    
-    for entry in data:
-        if 'mood' in entry and 'date' in entry:
-            moods.append(entry['mood'])
-            dates.append(entry['date'])
+# Create Gradio interface
+def create_gradio_app():
+    app = PsychologyToolsApp()
     
-    if not moods:
-        return "Error: No valid mood entries found."
-    
-    # Calculate statistics
-    avg_mood = sum(moods) / len(moods)
-    mood_trend = "stable"
-    
-    if len(moods) > 3:
-        recent_avg = sum(moods[-3:]) / 3
-        older_avg = sum(moods[:-3]) / (len(moods) - 3)
-        
-        if recent_avg > older_avg + 0.5:
-            mood_trend = "improving"
-        elif recent_avg < older_avg - 0.5:
-            mood_trend = "declining"
-    
-    # Create report
-    report = "# 📊 Mood Tracking Analysis\n\n"
-    report += f"**Period**: {dates[0]} to {dates[-1]}\n"
-    report += f"**Entries analyzed**: {len(moods)}\n"
-    report += f"**Average mood**: {avg_mood:.1f}/10\n"
-    report += f"**Trend**: {mood_trend}\n\n"
-    
-    report += "## 📈 Pattern Analysis:\n"
-    
-    # Check for low mood periods
-    low_mood_days = sum(1 for m in moods if m < 5)
-    if low_mood_days > len(moods) * 0.3:
-        report += "⚠️ **Concern**: Low mood detected in >30% of entries\n"
-        report += "   → Consider professional evaluation for depression\n\n"
-    
-    # Check volatility
-    if len(moods) > 2:
-        volatility = sum(abs(moods[i] - moods[i-1]) for i in range(1, len(moods))) / (len(moods) - 1)
-        if volatility > 2:
-            report += "🎢 **High mood volatility detected**\n"
-            report += "   → May benefit from emotion regulation skills\n\n"
-    
-    report += "## 💡 Recommendations:\n"
-    
-    if mood_trend == "declining":
-        report += "- 🚨 Schedule appointment with mental health professional\n"
-        report += "- 🧘 Increase self-care activities\n"
-        report += "- 👥 Reach out to support network\n"
-    elif mood_trend == "improving":
-        report += "- ✅ Continue current strategies\n"
-        report += "- 📝 Document what's working well\n"
-        report += "- 🎯 Set new wellness goals\n"
-    else:
-        report += "- 📊 Continue tracking for more data\n"
-        report += "- 🔍 Look for mood triggers\n"
-        report += "- 💪 Maintain healthy routines\n"
-    
-    report += f"\n*Analysis completed: {datetime.datetime.now().strftime('%m/%d/%Y %H:%M')}*"
-    
-    return report
+    with gr.Blocks(title="Psychology Tools Suite", theme=gr.themes.Soft()) as demo:
+        gr.Markdown("# 🧠 Psychology Tools Suite")
+        gr.Markdown("AI-powered tools for mental health professionals")
+        
+        with gr.Tabs():
+            # Text Analysis Tab
+            with gr.TabItem("📝 Text Analysis"):
+                gr.Markdown("### Analyze emotional content from text")
+                text_input = gr.Textbox(
+                    label="Enter patient's text",
+                    placeholder="Type or paste text here...",
+                    lines=5
+                )
+                context_input = gr.Dropdown(
+                    choices=["therapy", "assessment", "crisis"],
+                    value="therapy",
+                    label="Context"
+                )
+                analyze_btn = gr.Button("Analyze Emotions", variant="primary")
+                analysis_output = gr.Markdown()
+                
+                analyze_btn.click(
+                    fn=app.speech_emotion_analyzer,
+                    inputs=[text_input, context_input],
+                    outputs=analysis_output
+                )
+            
+            # Worksheet Creator Tab
+            with gr.TabItem("📋 Worksheet Creator"):
+                gr.Markdown("### Create therapy worksheets")
+                therapy_type = gr.Dropdown(
+                    choices=["cbt", "dbt", "mindfulness"],
+                    value="cbt",
+                    label="Therapy Type"
+                )
+                age_input = gr.Number(
+                    label="Client Age",
+                    value=25,
+                    minimum=5,
+                    maximum=100
+                )
+                issue_input = gr.Textbox(
+                    label="Main Issue",
+                    placeholder="e.g., anxiety, depression, stress"
+                )
+                create_btn = gr.Button("Create Worksheet", variant="primary")
+                worksheet_output = gr.Markdown()
+                
+                create_btn.click(
+                    fn=app.create_therapy_worksheet,
+                    inputs=[therapy_type, age_input, issue_input],
+                    outputs=worksheet_output
+                )
+            
+            # Mood Analysis Tab
+            with gr.TabItem("📊 Mood Tracker"):
+                gr.Markdown("### Analyze mood tracking data")
+                gr.Markdown("Enter data in JSON format: `[{\"date\": \"2024-01-01\", \"mood\": 7}, ...]`")
+                mood_input = gr.Textbox(
+                    label="Mood Data (JSON)",
+                    placeholder='[{"date": "2024-01-01", "mood": 7}, {"date": "2024-01-02", "mood": 6}]',
+                    lines=5
+                )
+                analyze_mood_btn = gr.Button("Analyze Mood Patterns", variant="primary")
+                mood_output = gr.Markdown()
+                
+                analyze_mood_btn.click(
+                    fn=app.mood_tracker_analyzer,
+                    inputs=mood_input,
+                    outputs=mood_output
+                )
+            
+            # Audio Analysis Tab (with Whisper)
+            with gr.TabItem("🎤 Audio Analysis"):
+                gr.Markdown("### Transcribe and analyze audio sessions")
+                api_key_input = gr.Textbox(
+                    label="OpenAI API Key",
+                    placeholder="sk-...",
+                    type="password"
+                )
+                audio_input = gr.Audio(
+                    label="Upload Audio File",
+                    type="file"
+                )
+                transcribe_btn = gr.Button("Transcribe & Analyze", variant="primary")
+                audio_output = gr.Markdown()
+                
+                transcribe_btn.click(
+                    fn=app.transcribe_audio_with_whisper,
+                    inputs=[audio_input, api_key_input],
+                    outputs=audio_output
+                )
+        
+        # Examples
+        gr.Examples(
+            examples=[
+                ["I feel so sad and hopeless. Nothing seems to matter anymore.", "therapy"],
+                ["I'm constantly worried about everything. My heart races and I can't sleep.", "therapy"],
+                ["I'm doing much better. Feeling more positive and energetic.", "therapy"]
+            ],
+            inputs=[text_input, context_input]
+        )
+    
+    return demo
+
+# Launch the app
+if __name__ == "__main__":
+    demo = create_gradio_app()
+    demo.launch(share=True)