Update app.py

app.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 """
 Hybrid AI Assistant - General Purpose + Healthcare Billing Expert
-A ChatGPT-style assistant using Gradio ChatInterface for simplicity
+Enhanced with Emotional UI and Voice Input
 """
 
 import os
@@ -9,7 +9,7 @@ import sys
 import json
 import logging
 import re
-from typing import Dict, Optional, Tuple, List, Any, Iterator
+from typing import Dict, Optional, Tuple, List, Any
 from dataclasses import dataclass, field
 from enum import Enum
 import requests
@@ -40,6 +40,19 @@ class ConversationContext:
     messages: List[Dict[str, str]] = field(default_factory=list)
     detected_codes: List[str] = field(default_factory=list)
     last_topic: Optional[str] = None
+    current_sentiment: str = "neutral"
+    sentiment_history: List[str] = field(default_factory=list)
+
+class SentimentType(Enum):
+    VERY_POSITIVE = "very_positive"
+    POSITIVE = "positive"
+    NEUTRAL = "neutral"
+    NEGATIVE = "negative"
+    VERY_NEGATIVE = "very_negative"
+    ANXIOUS = "anxious"
+    FRUSTRATED = "frustrated"
+    EXCITED = "excited"
+    CONFUSED = "confused"
 
 # ============= Healthcare Billing Database =============
 
@@ -181,12 +194,50 @@ class BillingCodesDB:
         
         return found_codes
 
+# ============= Sentiment Analysis =============
+
+class SentimentAnalyzer:
+    def __init__(self):
+        self.positive_words = ['great', 'awesome', 'excellent', 'fantastic', 'wonderful', 'amazing', 'perfect', 'love', 'happy', 'excited', 'thank', 'thanks', 'good', 'nice', 'brilliant', 'outstanding']
+        self.negative_words = ['terrible', 'awful', 'horrible', 'bad', 'worst', 'hate', 'frustrated', 'angry', 'sad', 'disappointed', 'upset', 'confused', 'difficult', 'problem', 'issue', 'error', 'wrong']
+        self.anxious_words = ['worried', 'concerned', 'nervous', 'anxious', 'scared', 'afraid', 'stress', 'panic', 'uncertain', 'unsure']
+        self.excited_words = ['excited', 'thrilled', 'amazing', 'wow', 'incredible', 'fantastic', 'brilliant', 'awesome']
+        
+    def analyze_sentiment(self, text: str) -> SentimentType:
+        text_lower = text.lower()
+        
+        positive_count = sum(1 for word in self.positive_words if word in text_lower)
+        negative_count = sum(1 for word in self.negative_words if word in text_lower)
+        anxious_count = sum(1 for word in self.anxious_words if word in text_lower)
+        excited_count = sum(1 for word in self.excited_words if word in text_lower)
+        
+        # Check for question marks (confusion indicator)
+        question_marks = text.count('?')
+        exclamation_marks = text.count('!')
+        
+        # Determine sentiment
+        if excited_count > 0 or exclamation_marks > 1:
+            return SentimentType.EXCITED
+        elif anxious_count > 0:
+            return SentimentType.ANXIOUS
+        elif question_marks > 1 and negative_count > 0:
+            return SentimentType.CONFUSED
+        elif negative_count > positive_count and negative_count > 1:
+            return SentimentType.VERY_NEGATIVE if negative_count > 2 else SentimentType.NEGATIVE
+        elif positive_count > negative_count and positive_count > 1:
+            return SentimentType.VERY_POSITIVE if positive_count > 2 else SentimentType.POSITIVE
+        elif 'frustrated' in text_lower or 'frustrating' in text_lower:
+            return SentimentType.FRUSTRATED
+        else:
+            return SentimentType.NEUTRAL
+
 # ============= AI Assistant Class =============
 
 class HybridAIAssistant:
     def __init__(self):
         self.api_key = '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'
         self.billing_db = BillingCodesDB()
+        self.sentiment_analyzer = SentimentAnalyzer()
         self.context = ConversationContext()
         
         self.headers = {
@@ -239,15 +290,39 @@ class HybridAIAssistant:
         else:
             return self.get_general_response(message, billing_context=True)
     
+    def get_empathetic_response_prefix(self, sentiment: SentimentType) -> str:
+        """Generate empathetic response based on sentiment"""
+        prefixes = {
+            SentimentType.VERY_POSITIVE: "I'm so glad to hear your enthusiasm! 🌟 ",
+            SentimentType.POSITIVE: "That's wonderful! 😊 ",
+            SentimentType.EXCITED: "I can feel your excitement! 🎉 ",
+            SentimentType.ANXIOUS: "I understand this might be causing some concern. Let me help ease your worries. 🤗 ",
+            SentimentType.FRUSTRATED: "I can sense your frustration, and I'm here to help make this easier for you. 💙 ",
+            SentimentType.CONFUSED: "No worries, I'm here to clear things up for you! 🧠 ",
+            SentimentType.NEGATIVE: "I hear that you're having some difficulties. Let me help you with that. 💚 ",
+            SentimentType.VERY_NEGATIVE: "I'm really sorry you're going through this. I'm here to support you. ❤️ ",
+            SentimentType.NEUTRAL: ""
+        }
+        return prefixes.get(sentiment, "")
+    
     def get_general_response(self, message: str, billing_context: bool = False) -> str:
         """Get response from OpenRouter API for general queries"""
         
-        # Prepare system prompt
-        system_prompt = """You are a helpful, friendly AI assistant with expertise in healthcare billing codes. 
+        # Analyze sentiment
+        sentiment = self.sentiment_analyzer.analyze_sentiment(message)
+        self.context.current_sentiment = sentiment.value
+        self.context.sentiment_history.append(sentiment.value)
+        
+        # Keep only last 10 sentiments
+        if len(self.context.sentiment_history) > 10:
+            self.context.sentiment_history = self.context.sentiment_history[-10:]
+        
+        # Prepare system prompt with empathy
+        system_prompt = """You are a helpful, friendly, and empathetic AI assistant with expertise in healthcare billing codes. 
         You can assist with any topic - from casual conversation to complex questions. 
         When discussing medical billing codes, you provide accurate, detailed information.
-        Be conversational, helpful, and engaging. Use emojis occasionally to be friendly.
-        Keep responses concise but informative."""
+        Be conversational, helpful, and engaging. Show empathy and understanding.
+        Adapt your tone based on the user's emotional state - be more supportive if they seem frustrated or anxious."""
         
         if billing_context:
             system_prompt += "\nThe user is asking about medical billing. Provide helpful information even if you don't have specific code details."
@@ -280,6 +355,11 @@ class HybridAIAssistant:
                 result = response.json()
                 ai_response = result['choices'][0]['message']['content']
                 
+                # Add empathetic prefix based on sentiment
+                empathy_prefix = self.get_empathetic_response_prefix(sentiment)
+                if empathy_prefix:
+                    ai_response = empathy_prefix + ai_response
+                
                 # Update context
                 self.context.messages.append({'role': 'user', 'content': message})
                 self.context.messages.append({'role': 'assistant', 'content': ai_response})
@@ -299,27 +379,32 @@ class HybridAIAssistant:
     
     def get_fallback_response(self, message: str) -> str:
         """Fallback responses when API fails"""
+        sentiment = self.sentiment_analyzer.analyze_sentiment(message)
+        empathy_prefix = self.get_empathetic_response_prefix(sentiment)
+        
         fallbacks = [
             "I'm having trouble connecting right now, but I'm still here to help! Could you rephrase your question?",
             "Let me think about that differently. What specific aspect would you like to know more about?",
             "That's an interesting question! While I process that, is there anything specific you'd like to explore?",
             "I'm here to help! Could you provide a bit more detail about what you're looking for?"
         ]
-        return random.choice(fallbacks)
+        return empathy_prefix + random.choice(fallbacks)
     
-    def process_message(self, message: str) -> str:
-        """Main method to process any message"""
+    def process_message(self, message: str) -> Tuple[str, str]:
+        """Main method to process any message and return response with sentiment"""
         if not message.strip():
-            return "Feel free to ask me anything! I can help with general questions or healthcare billing codes. 😊"
+            return "Feel free to ask me anything! I can help with general questions or healthcare billing codes. 😊", "neutral"
         
         # Detect intent
         intent = self.detect_intent(message)
         
         # Route to appropriate handler
         if intent['is_billing'] and intent['codes_found']:
-            return self.handle_billing_query(message, intent['codes_found'])
+            response = self.handle_billing_query(message, intent['codes_found'])
         else:
-            return self.get_general_response(message, billing_context=intent['is_billing'])
+            response = self.get_general_response(message, billing_context=intent['is_billing'])
+        
+        return response, self.context.current_sentiment
     
     def reset_context(self):
         """Reset conversation context"""
@@ -335,10 +420,21 @@ def respond(message, history):
     if not message.strip():
         return "Feel free to ask me anything! I can help with general questions or healthcare billing codes. 😊"
     
-    # Process message
-    response = assistant.process_message(message)
+    # Process message and get sentiment
+    response, sentiment = assistant.process_message(message)
+    
+    # Update UI based on sentiment (this will be handled by JavaScript)
     return response
 
+def process_voice_input(audio):
+    """Process voice input and return text"""
+    if audio is None:
+        return "No audio received. Please try again."
+    
+    # For now, return a placeholder message
+    # In a real implementation, you'd use speech recognition here
+    return "Voice input processed! (Speech recognition would be implemented here)"
+
 def reset_chat():
     """Reset the conversation context"""
     assistant.reset_context()
@@ -348,22 +444,22 @@ def reset_chat():
 
 examples = [
     "What is healthcare billing code A0429?",
-    "Can you explain CPT code 99213 in detail?",
+    "Can you explain CPT code 99213 in detail?", 
     "Tell me about DRG 470",
+    "I'm feeling frustrated with this billing issue",
+    "This is confusing, can you help me understand?",
+    "Thank you so much! This is exactly what I needed!",
     "How does artificial intelligence work?",
     "Give me a simple pasta recipe",
-    "Teach me Python basics",
-    "Write a short poem about nature",
-    "Help me write a professional email template",
-    "Give me creative story ideas"
+    "Write a short poem about nature"
 ]
 
 # ============= Create Interface =============
 
 def create_interface():
-    """Create the Gradio ChatInterface"""
+    """Create the Gradio ChatInterface with Emotional UI and Voice Input"""
     
-    # Enhanced CSS with modern design improvements
+    # Enhanced CSS with emotional UI and voice features
     custom_css = """
     /* Global Styles */
     .gradio-container {
@@ -373,9 +469,20 @@ def create_interface():
         background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%) !important;
         min-height: 100vh !important;
         padding: 1rem !important;
+        transition: all 0.5s ease !important;
     }
     
-    /* Enhanced Header */
+    /* Emotional UI Color Schemes */
+    .sentiment-positive { background: linear-gradient(135deg, #84fab0 0%, #8fd3f4 100%) !important; }
+    .sentiment-very-positive { background: linear-gradient(135deg, #a8edea 0%, #fed6e3 100%) !important; }
+    .sentiment-negative { background: linear-gradient(135deg, #d299c2 0%, #fef9d7 100%) !important; }
+    .sentiment-very-negative { background: linear-gradient(135deg, #ff9a9e 0%, #fecfef 100%) !important; }
+    .sentiment-anxious { background: linear-gradient(135deg, #ffecd2 0%, #fcb69f 100%) !important; }
+    .sentiment-frustrated { background: linear-gradient(135deg, #ff8a80 0%, #ffad80 100%) !important; }
+    .sentiment-excited { background: linear-gradient(135deg, #ffd89b 0%, #19547b 100%) !important; }
+    .sentiment-confused { background: linear-gradient(135deg, #a8caba 0%, #5d4e75 100%) !important; }
+    
+    /* Enhanced Header with Mood Indicator */
     .header-text {
         text-align: center;
         background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
@@ -386,6 +493,7 @@ def create_interface():
         box-shadow: 0 20px 40px rgba(102, 126, 234, 0.3);
         position: relative;
         overflow: hidden;
+        transition: all 0.5s ease;
     }
     
     .header-text::before {
@@ -404,6 +512,34 @@ def create_interface():
         50% { opacity: 0.6; }
     }
     
+    /* Mood Indicator */
+    .mood-indicator {
+        position: absolute;
+        top: 1rem;
+        right: 1rem;
+        width: 60px;
+        height: 60px;
+        border-radius: 50%;
+        background: rgba(255,255,255,0.2);
+        backdrop-filter: blur(10px);
+        display: flex;
+        align-items: center;
+        justify-content: center;
+        font-size: 24px;
+        transition: all 0.5s ease;
+        animation: breathe 3s ease-in-out infinite;
+    }
+    
+    @keyframes breathe {
+        0%, 100% { transform: scale(1); }
+        50% { transform: scale(1.05); }
+    }
+    
+    .mood-positive { background: rgba(132, 250, 176, 0.3) !important; }
+    .mood-negative { background: rgba(255, 154, 158, 0.3) !important; }
+    .mood-anxious { background: rgba(255, 236, 210, 0.3) !important; }
+    .mood-excited { background: rgba(255, 216, 155, 0.3) !important; }
+    
     .header-text h1 {
         margin: 0;
         font-size: 3rem;
@@ -438,7 +574,39 @@ def create_interface():
         to { box-shadow: 0 0 20px rgba(255,255,255,0.6); }
     }
     
-    /* Chat Interface Styling */
+    /* Voice Input Button */
+    .voice-btn {
+        background: linear-gradient(135deg, #ff6b6b 0%, #ee5a52 100%) !important;
+        color: white !important;
+        border: none !important;
+        border-radius: 50% !important;
+        width: 60px !important;
+        height: 60px !important;
+        font-size: 24px !important;
+        margin: 0.5rem !important;
+        transition: all 0.3s cubic-bezier(0.175, 0.885, 0.32, 1.275) !important;
+        box-shadow: 0 4px 15px rgba(255, 107, 107, 0.3) !important;
+        position: relative;
+        overflow: hidden;
+    }
+    
+    .voice-btn:hover {
+        transform: scale(1.1) !important;
+        box-shadow: 0 8px 25px rgba(255, 107, 107, 0.5) !important;
+    }
+    
+    .voice-btn.recording {
+        animation: recordPulse 1s ease-in-out infinite !important;
+        background: linear-gradient(135deg, #ff3030 0%, #ff1010 100%) !important;
+    }
+    
+    @keyframes recordPulse {
+        0% { box-shadow: 0 0 0 0 rgba(255, 107, 107, 0.7); }
+        70% { box-shadow: 0 0 0 20px rgba(255, 107, 107, 0); }
+        100% { box-shadow: 0 0 0 0 rgba(255, 107, 107, 0); }
+    }
+    
+    /* Chat Interface Styling with Emotional Feedback */
     .gradio-chatinterface {
         background: white !important;
         border-radius: 20px !important;
@@ -446,6 +614,17 @@ def create_interface():
         padding: 2rem !important;
         margin: 1rem 0 !important;
         backdrop-filter: blur(10px) !important;
+        transition: all 0.5s ease !important;
+    }
+    
+    .gradio-chatinterface.emotional-positive {
+        border: 2px solid rgba(132, 250, 176, 0.5) !important;
+        box-shadow: 0 25px 50px rgba(132, 250, 176, 0.2) !important;
+    }
+    
+    .gradio-chatinterface.emotional-negative {
+        border: 2px solid rgba(255, 154, 158, 0.5) !important;
+        box-shadow: 0 25px 50px rgba(255, 154, 158, 0.2) !important;
     }
     
     /* Enhanced Buttons */
@@ -539,6 +718,40 @@ def create_interface():
         letter-spacing: 1px;
     }
     
+    /* Empathy Animations */
+    @keyframes empathyPulse {
+        0%, 100% { transform: scale(1); }
+        50% { transform: scale(1.02); }
+    }
+    
+    @keyframes supportGlow {
+        0%, 100% { box-shadow: 0 0 10px rgba(102, 126, 234, 0.3); }
+        50% { box-shadow: 0 0 20px rgba(102, 126, 234, 0.6); }
+    }
+    
+    .empathy-support {
+        animation: empathyPulse 2s ease-in-out infinite, supportGlow 3s ease-in-out infinite;
+    }
+    
+    /* Voice Recognition Feedback */
+    .voice-feedback {
+        position: fixed;
+        bottom: 2rem;
+        right: 2rem;
+        background: rgba(102, 126, 234, 0.9);
+        color: white;
+        padding: 1rem 1.5rem;
+        border-radius: 50px;
+        backdrop-filter: blur(10px);
+        z-index: 1000;
+        animation: slideInRight 0.3s ease-out;
+    }
+    
+    @keyframes slideInRight {
+        from { transform: translateX(100%); opacity: 0; }
+        to { transform: translateX(0); opacity: 1; }
+    }
+    
     /* Enhanced Accordion */
     .gradio-accordion {
         background: rgba(255,255,255,0.9) !important;
@@ -564,6 +777,12 @@ def create_interface():
         padding: 1.5rem;
         border: 1px solid rgba(255,255,255,0.3);
         box-shadow: 0 8px 25px rgba(0,0,0,0.1);
+        transition: all 0.3s ease;
+    }
+    
+    .feature-card:hover {
+        transform: translateY(-5px);
+        box-shadow: 0 15px 35px rgba(0,0,0,0.15);
     }
     
     .feature-icon {
@@ -612,6 +831,18 @@ def create_interface():
             grid-template-columns: 1fr;
             gap: 1rem;
         }
+        
+        .mood-indicator {
+            width: 50px;
+            height: 50px;
+            font-size: 20px;
+        }
+        
+        .voice-btn {
+            width: 50px !important;
+            height: 50px !important;
+            font-size: 20px !important;
+        }
     }
     
     /* Loading Animation */
@@ -625,71 +856,20 @@ def create_interface():
         background: linear-gradient(90deg, #f0f0f0 25%, #e0e0e0 50%, #f0f0f0 75%);
         background-size: 400% 100%;
     }
-    """
-    
-    with gr.Blocks(css=custom_css, title="AI Assistant + Healthcare Billing Expert") as demo:
-        # Header
-        gr.HTML("""
-            <div class="header-text">
-                <h1>🤖 AI Assistant <span style="font-size: 0.6em; background: rgba(255,255,255,0.2); padding: 4px 12px; border-radius: 16px;">PLUS</span></h1>
-                <p>Your intelligent companion for any question + Healthcare Billing Expert</p>
-            </div>
-        """)
-        
-        # Main Chat Interface
-        chat_interface = gr.ChatInterface(
-            fn=respond,
-            examples=examples,
-            cache_examples=False
-        )
-        
-        # Additional controls
-        with gr.Row():
-            with gr.Column(scale=1):
-                reset_context_btn = gr.Button("🔄 Reset Context", elem_classes="reset-btn", size="sm")
-            with gr.Column(scale=3):
-                gr.HTML("")  # Spacer
-            with gr.Column(scale=1):
-                gr.HTML("""
-                    <div style='text-align: right; color: #718096; font-size: 12px; margin-top: 0.5rem;'>
-                        Powered by GPT-3.5 Turbo<br>
-                        Healthcare Billing Database
-                    </div>
-                """)
-        
-        # Info section
-        with gr.Accordion("ℹ️ About This Assistant", open=False):
-            gr.HTML("""
-                <div style="padding: 1rem; background: #f7fafc; border-radius: 8px; margin: 0.5rem 0;">
-                    <h4 style="color: #2d3748; margin-top: 0;">🏥 Healthcare Billing Expert</h4>
-                    <p style="color: #4a5568; margin-bottom: 1rem;">I'm specialized in healthcare billing codes including:</p>
-                    <ul style="color: #4a5568; margin: 0.5rem 0;">
-                        <li><strong>CPT Codes</strong> - Current Procedural Terminology</li>
-                        <li><strong>HCPCS</strong> - Healthcare Common Procedure Coding System</li>
-                        <li><strong>ICD-10</strong> - International Classification of Diseases</li>
-                        <li><strong>DRG</strong> - Diagnosis-Related Groups</li>
-                    </ul>
-                    
-                    <h4 style="color: #2d3748;">💬 General AI Assistant</h4>
-                    <p style="color: #4a5568; margin: 0;">I can also help with general questions, writing, coding, learning, and creative tasks!</p>
-                </div>
-            """)
-        
-        # Connect reset button
-        reset_context_btn.click(
-            fn=reset_chat,
-            outputs=chat_interface.chatbot
-        )
-    
-    return demo
-
-# ============= Launch =============
-
-if __name__ == "__main__":
-    app = create_interface()
-    app.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        show_error=True
-    )
+    
+    /* Sentiment-based message styling */
+    .message-positive {
+        border-left: 4px solid #84fab0 !important;
+        background: linear-gradient(135deg, rgba(132, 250, 176, 0.1) 0%, rgba(143, 211, 244, 0.1) 100%) !important;
+    }
+    
+    .message-negative {
+        border-left: 4px solid #ff9a9e !important;
+        background: linear-gradient(135deg, rgba(255, 154, 158, 0.1) 0%, rgba(254, 207, 239, 0.1) 100%) !important;
+    }
+    
+    .message-anxious {
+        border-left: 4px solid #ffecd2 !important;
+        background: linear-gradient(135deg, rgba(255, 236, 210, 0.1) 0%, rgba(252, 182, 159, 0.1) 100%) !important;
+    }
+    """