Force CPU-only mode and disable GPU completely

app.py

@@ -7,18 +7,19 @@ from utils.log_manager import LogManager
 
 def setup_environment():
     """Setup environment variables and device configuration"""
-    # Force CPU usage if GPU is not available
-    if not torch.cuda.is_available():
-        os.environ["CUDA_VISIBLE_DEVICES"] = ""
-        device = "cpu"
-    else:
-        device = "cuda"
-    
-    # Set transformers to use CPU
+    # Force CPU usage and disable GPU
+    os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
     os.environ["TRANSFORMERS_OFFLINE"] = "1"
-    os.environ["TORCH_DEVICE"] = device
+    os.environ["TORCH_DEVICE"] = "cpu"
+    
+    # Disable CUDA initialization
+    torch.cuda.is_available = lambda: False
     
-    return device
+    # Set PyTorch to use CPU
+    if hasattr(torch, 'set_default_tensor_type'):
+        torch.set_default_tensor_type('torch.FloatTensor')
+    
+    return "cpu"
 
 def main():
     # Initialize logging
@@ -27,10 +28,13 @@ def main():
     logger.info("Starting Mental Wellness Platform")
     
     try:
-        # Setup environment
+        # Setup environment before anything else
         device = setup_environment()
         logger.info(f"Using device: {device}")
         
+        # Disable GPU memory allocation
+        torch.cuda.empty_cache()
+        
         # Load configuration
         config = load_config()
         logger.info("Configuration loaded successfully")
@@ -44,7 +48,8 @@ def main():
             server_name="0.0.0.0",
             server_port=7860,
             show_error=True,  # Show detailed error messages
-            root_path=""  # Empty root path for Spaces
+            root_path="",  # Empty root path for Spaces
+            quiet=True  # Reduce logging noise
         )
         
     except Exception as e:

interface/app.py

@@ -5,6 +5,11 @@ import logging
 from utils.log_manager import LogManager
 from utils.analytics_logger import AnalyticsLogger
 
+# Force CPU-only mode
+torch.cuda.is_available = lambda: False
+if hasattr(torch, 'set_default_tensor_type'):
+    torch.set_default_tensor_type('torch.FloatTensor')
+
 class WellnessInterface:
     def __init__(self, config):
         self.config = config
@@ -12,10 +17,11 @@ class WellnessInterface:
         self.logger = self.log_manager.get_agent_logger("interface")
         self.analytics = AnalyticsLogger()
         
-        # Set device configuration
-        self.device = os.getenv("TORCH_DEVICE", "cpu")
-        self.logger.info(f"Using device: {self.device}")
+        # Ensure CPU-only operation
+        self.device = "cpu"
+        self.logger.info("Using CPU-only mode")
         
+        # Initialize interface
         self.setup_interface()
         
     def setup_interface(self):
@@ -105,7 +111,7 @@ class WellnessInterface:
                     ],
                     outputs=[
                         self.chatbot,
-                        self.text_input  # Add text_input to clear it after sending
+                        self.text_input
                     ],
                     api_name="chat"
                 )
@@ -135,7 +141,7 @@ class WellnessInterface:
                     ],
                     outputs=[
                         self.chatbot,
-                        self.text_input  # Add text_input to clear it after sending
+                        self.text_input
                     ]
                 )
                 
@@ -149,7 +155,7 @@ class WellnessInterface:
         """Process user input from various sources"""
         try:
             if not text and not audio and not image:
-                return history, ""  # Return empty string to keep text input unchanged
+                return history, ""
                 
             # Log the interaction start
             self.analytics.log_user_interaction(