Fix Gradio compatibility issues and update interface

interface/app.py

@@ -1,198 +1,156 @@
 import gradio as gr
-from typing import Dict, List
-import os
+import logging
+from utils.log_manager import LogManager
+from utils.analytics_logger import AnalyticsLogger
 
 class WellnessInterface:
-    def __init__(self, config: Dict):
+    def __init__(self, config):
         self.config = config
-        self.interface = None
+        self.log_manager = LogManager()
+        self.logger = self.log_manager.get_agent_logger("interface")
+        self.analytics = AnalyticsLogger()
         self.setup_interface()
         
     def setup_interface(self):
-        """Set up the Gradio interface"""
-        with gr.Blocks(theme=gr.themes.Soft()) as self.interface:
-            gr.Markdown("# Mental Wellness Support Platform")
-            gr.Markdown("Welcome to your AI-powered mental wellness support system. "
-                       "Please note that this is not a substitute for professional mental health care.")
-            
-            with gr.Row():
-                with gr.Column(scale=3):
-                    # Chat interface
-                    self.chatbot = gr.Chatbot(
-                        label="Conversation",
-                        height=400
-                    )
-                    with gr.Row():
-                        self.msg = gr.Textbox(
-                            label="Type your message",
-                            placeholder="Share what's on your mind...",
-                            scale=4
-                        )
-                        self.send = gr.Button("Send", scale=1)
-                        
-                    # Voice input
-                    self.audio_input = gr.Audio(
-                        label="Voice Input",
-                        source="microphone",
-                        type="filepath"
-                    )
-                    
-                with gr.Column(scale=1):
-                    # Wellness tools
-                    with gr.Tab("Tools"):
-                        gr.Markdown("### Wellness Tools")
-                        self.meditation_btn = gr.Button("Start Meditation")
-                        self.assessment_btn = gr.Button("Mental Health Check-in")
-                        self.resources_btn = gr.Button("View Resources")
-                        
-                    # Session analytics
-                    with gr.Tab("Analytics"):
-                        self.analytics_display = gr.JSON(
-                            label="Session Analytics",
-                            value={}
-                        )
-                        self.update_analytics_btn = gr.Button("Update Analytics")
-                        
-            # File upload for image/video
-            with gr.Row():
-                self.file_upload = gr.File(
-                    label="Upload Image/Video",
-                    file_types=["image", "video"]
-                )
-                
-            # Emergency resources
-            with gr.Row():
-                gr.Markdown("""
-                ### Emergency Resources
-                If you're experiencing a mental health emergency:
-                - Emergency Services: 911 (US)
-                - National Crisis Hotline: 988
-                - Crisis Text Line: Text HOME to 741741
-                """)
-                
-            # Set up event handlers
-            self.send.click(
-                fn=self._handle_text_input,
-                inputs=[self.msg, self.chatbot],
-                outputs=[self.msg, self.chatbot]
+        """Setup the Gradio interface components"""
+        self.logger.info("Setting up interface components")
+        
+        try:
+            # Chat interface
+            self.chatbot = gr.Chatbot(
+                label="Mental Wellness Assistant",
+                type="messages",  # Using new message format
+                height=400
             )
             
-            self.audio_input.change(
-                fn=self._handle_voice_input,
-                inputs=[self.audio_input, self.chatbot],
-                outputs=[self.chatbot]
+            # Text input
+            self.text_input = gr.Textbox(
+                label="Type your message",
+                placeholder="Enter your message here...",
+                lines=2
             )
             
-            self.file_upload.change(
-                fn=self._handle_file_upload,
-                inputs=[self.file_upload, self.chatbot],
-                outputs=[self.chatbot]
+            # Audio input
+            self.audio_input = gr.Audio(
+                label="Voice Input",
+                type="filepath",  # Using filepath instead of source
+                format="wav"
             )
             
-            self.update_analytics_btn.click(
-                fn=self._update_analytics,
-                inputs=[],
-                outputs=[self.analytics_display]
+            # Image input
+            self.image_input = gr.Image(
+                label="Image Upload",
+                type="filepath"
             )
             
-            # Wellness tool handlers
-            self.meditation_btn.click(
-                fn=self._start_meditation,
-                inputs=[self.chatbot],
+            # Control buttons
+            self.submit_btn = gr.Button("Send")
+            self.clear_btn = gr.Button("Clear Chat")
+            self.emergency_btn = gr.Button("Emergency Help", variant="secondary")
+            
+            # Setup interface layout
+            self.interface = gr.Interface(
+                fn=self.process_input,
+                inputs=[
+                    self.text_input,
+                    self.audio_input,
+                    self.image_input,
+                    self.chatbot
+                ],
+                outputs=[self.chatbot],
+                title="Mental Wellness Support",
+                description="A safe space for mental health support and guidance.",
+                theme=self.config["INTERFACE_CONFIG"]["theme"],
+                allow_flagging="never"
+            )
+            
+            # Setup event handlers
+            self.submit_btn.click(
+                fn=self.process_input,
+                inputs=[
+                    self.text_input,
+                    self.audio_input,
+                    self.image_input,
+                    self.chatbot
+                ],
                 outputs=[self.chatbot]
             )
             
-            self.assessment_btn.click(
-                fn=self._start_assessment,
-                inputs=[self.chatbot],
+            self.clear_btn.click(
+                fn=self.clear_chat,
+                inputs=[],
                 outputs=[self.chatbot]
             )
             
-            self.resources_btn.click(
-                fn=self._show_resources,
-                inputs=[self.chatbot],
+            self.emergency_btn.click(
+                fn=self.emergency_help,
+                inputs=[],
                 outputs=[self.chatbot]
             )
             
-    def _handle_text_input(self, message: str, history: list) -> tuple:
-        """Handle text input from user"""
-        if message:
-            # For now, just echo the message
-            response = f"You said: {message}"
-            history.append((message, response))
-            return "", history
-        return message, history
-        
-    def _handle_voice_input(self, audio_path: str, history: list) -> list:
-        """Handle voice input from user"""
-        if audio_path:
-            response = "Voice message received! (Voice processing coming soon)"
-            history.append(("🎤 Voice message", response))
-        return history
-        
-    def _handle_file_upload(self, file: gr.File, history: list) -> list:
-        """Handle file upload from user"""
-        if file:
-            response = f"Received file: {file.name} (Media processing coming soon)"
-            history.append((f"📎 Uploaded: {file.name}", response))
-        return history
-        
-    def _update_analytics(self) -> Dict:
-        """Update analytics display"""
-        return {
-            "sessions": 1,
-            "messages": len(self.chatbot.value) if self.chatbot.value else 0,
-            "status": "Demo Mode"
-        }
-        
-    def _start_meditation(self, history: list) -> list:
-        """Start meditation session"""
-        response = """Let's begin a simple breathing exercise:
-        1. Find a comfortable position
-        2. Close your eyes or maintain a soft gaze
-        3. We'll practice deep breathing together
-        
-        Are you ready to start? (Reply 'yes' when ready)"""
-        history.append(("🧘 Started meditation session", response))
-        return history
-        
-    def _start_assessment(self, history: list) -> list:
-        """Start mental health assessment"""
-        response = """I'll help you with a quick mental health check-in.
-        We'll use standard screening tools to understand how you're feeling.
-        
-        Would you like to:
-        1. Check anxiety levels (GAD-7)
-        2. Check mood/depression (PHQ-9)
-        3. General well-being assessment
-        
-        Please choose a number (1-3)."""
-        history.append(("📋 Started assessment", response))
-        return history
-        
-    def _show_resources(self, history: list) -> list:
-        """Show mental health resources"""
-        response = """Here are some helpful resources:
-        
-        📱 Quick Tools:
-        - Breathing exercises
-        - Grounding techniques
-        - Meditation guides
+            self.logger.info("Interface setup completed successfully")
+            
+        except Exception as e:
+            self.logger.error(f"Error setting up interface: {str(e)}")
+            raise
+            
+    def process_input(self, text, audio, image, history):
+        """Process user input from various sources"""
+        try:
+            # Log the interaction start
+            self.analytics.log_user_interaction(
+                user_id="anonymous",
+                interaction_type="message",
+                agent_type="interface",
+                duration=0,
+                success=True,
+                details={"input_types": {
+                    "text": bool(text),
+                    "audio": bool(audio),
+                    "image": bool(image)
+                }}
+            )
+            
+            # Process the input (placeholder)
+            response = "I understand and I'm here to help. Could you tell me more about how you're feeling?"
+            
+            # Add to chat history using the new message format
+            history.append({"role": "user", "content": text if text else "Sent media"})
+            history.append({"role": "assistant", "content": response})
+            
+            return history
+            
+        except Exception as e:
+            self.logger.error(f"Error processing input: {str(e)}")
+            return history + [
+                {"role": "assistant", "content": "I apologize, but I encountered an error. Please try again."}
+            ]
+            
+    def clear_chat(self):
+        """Clear the chat history"""
+        self.logger.info("Clearing chat history")
+        return []
         
-        📚 Educational Resources:
-        - Understanding anxiety
-        - Stress management
-        - Building resilience
+    def emergency_help(self):
+        """Provide emergency help information"""
+        self.logger.info("Emergency help requested")
         
-        🤝 Support Services:
-        - Find local therapists
-        - Support groups
-        - Crisis hotlines
+        emergency_message = [
+            {"role": "assistant", "content": """
+            If you're experiencing a mental health emergency:
+            
+            🚨 Emergency Services: 911 (US)
+            🆘 National Crisis Hotline: 988
+            💭 Crisis Text Line: Text HOME to 741741
+            
+            These services are available 24/7 and are staffed by trained professionals.
+            Your life matters, and help is available.
+            """}
+        ]
         
-        Which would you like to learn more about?"""
-        history.append(("📚 Resource guide", response))
-        return history
+        return emergency_message
         
     def launch(self, **kwargs):
-        """Launch the Gradio interface"""
+        """Launch the interface"""
+        self.logger.info("Launching interface")
         self.interface.launch(**kwargs)