fix logging

backend/logging_utils.py

@@ -6,87 +6,70 @@ import os
 from typing import Optional
 
 # Configure logging
-log_dir = Path(__file__).parent.parent / "logs"
-log_dir.mkdir(exist_ok=True)
+# Use /tmp for logs in production (e.g. Hugging Face Spaces) or local logs dir in development
+log_dir = Path("/tmp/schematic_ai_logs") if os.environ.get("SPACE_ID") else Path(__file__).parent.parent / "logs"
+log_dir.mkdir(exist_ok=True, parents=True)
 
 # Configure file handler for general logs
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
     handlers=[
-        logging.FileHandler(log_dir / "app.log"),
-        logging.StreamHandler()
+        logging.StreamHandler()  # Always log to console
     ]
 )
 
-# Create a specific logger for category usage
-category_logger = logging.getLogger('category_usage')
-category_logger.setLevel(logging.INFO)
+# Only add file handler if we can write to the directory
+try:
+    if os.access(log_dir, os.W_OK):
+        file_handler = logging.FileHandler(log_dir / "app.log")
+        file_handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
+        logging.getLogger().addHandler(file_handler)
+except Exception as e:
+    logging.warning(f"Could not set up file logging: {e}")
 
-# Stats file for category usage
-STATS_FILE = log_dir / "category_stats.json"
-
-def load_category_stats():
-    """Load existing category usage statistics."""
+def log_category_usage(category: Optional[str] = None):
+    """Log the usage of a category."""
+    if not os.access(log_dir, os.W_OK):
+        logging.warning("Log directory is not writable, skipping category usage logging")
+        return
+        
+    stats_file = log_dir / "category_stats.json"
+    
     try:
-        if STATS_FILE.exists():
-            with open(STATS_FILE, 'r') as f:
-                return json.load(f)
+        if stats_file.exists():
+            with open(stats_file, 'r') as f:
+                stats = json.load(f)
+        else:
+            stats = {}
+        
+        # Initialize or increment category count
+        category = category or "default"
+        if category in stats:
+            stats[category] += 1
+        else:
+            stats[category] = 1
+            
+        # Save updated stats
+        with open(stats_file, 'w') as f:
+            json.dump(stats, f, indent=4)
+            
     except Exception as e:
-        logging.error(f"Error loading category stats: {e}")
-    return {}
+        logging.error(f"Error logging category usage: {e}")
 
-def save_category_stats(stats):
-    """Save category usage statistics."""
+def get_category_statistics():
+    """Get the usage statistics for all categories."""
+    if not os.access(log_dir, os.W_OK):
+        logging.warning("Log directory is not writable, cannot read category statistics")
+        return {}
+        
+    stats_file = log_dir / "category_stats.json"
+    
     try:
-        with open(STATS_FILE, 'w') as f:
-            json.dump(stats, f, indent=2)
+        if stats_file.exists():
+            with open(stats_file, 'r') as f:
+                return json.load(f)
+        return {}
     except Exception as e:
-        logging.error(f"Error saving category stats: {e}")
-
-def log_category_usage(category: Optional[str], endpoint: str, success: bool):
-    """Log category usage with detailed metrics."""
-    timestamp = datetime.now().isoformat()
-    
-    # Load existing stats
-    stats = load_category_stats()
-    
-    # Initialize category if not exists
-    category_key = category or "default"
-    if category_key not in stats:
-        stats[category_key] = {
-            "total_requests": 0,
-            "successful_requests": 0,
-            "failed_requests": 0,
-            "endpoints": {},
-            "last_used": None
-        }
-    
-    # Update category stats
-    stats[category_key]["total_requests"] += 1
-    if success:
-        stats[category_key]["successful_requests"] += 1
-    else:
-        stats[category_key]["failed_requests"] += 1
-    
-    # Update endpoint stats
-    if endpoint not in stats[category_key]["endpoints"]:
-        stats[category_key]["endpoints"][endpoint] = 0
-    stats[category_key]["endpoints"][endpoint] += 1
-    
-    # Update timestamp
-    stats[category_key]["last_used"] = timestamp
-    
-    # Save updated stats
-    save_category_stats(stats)
-    
-    # Log the event
-    category_logger.info(
-        f"Category usage - Category: {category_key}, "
-        f"Endpoint: {endpoint}, "
-        f"Success: {success}"
-    )
-
-def get_category_statistics():
-    """Get category usage statistics."""
-    return load_category_stats()
+        logging.error(f"Error reading category statistics: {e}")
+        return {}