Fix metrics and history with proper file permissions in Space environment

app.py

@@ -9,6 +9,7 @@ import json
 from typing import List, Dict
 import warnings
 import spaces
+import stat
 
 # Filter out warnings
 warnings.filterwarnings('ignore')
@@ -19,16 +20,35 @@ logger = logging.getLogger(__name__)
 
 # Environment variables
 HF_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
-MODEL_NAME = os.getenv("MODEL_NAME", "google/gemma-2-2b-it")  # Fixed model name
+MODEL_NAME = os.getenv("MODEL_NAME", "google/gemma-2-2b-it")
 
-# Create data directory for persistence
-DATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "data")
+# Create data directory for persistence in Space environment
+DATA_DIR = "/home/user/app/data"
 os.makedirs(DATA_DIR, exist_ok=True)
+# Ensure directory has write permissions
+os.chmod(DATA_DIR, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
+logger.info(f"Using data directory: {DATA_DIR}")
 
 # Cache and history files
 CACHE_DIR = os.path.join(DATA_DIR, "cache")
 HISTORY_FILE = os.path.join(DATA_DIR, "review_history.json")
 os.makedirs(CACHE_DIR, exist_ok=True)
+# Ensure cache directory has write permissions
+os.chmod(CACHE_DIR, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
+logger.info(f"Cache directory: {CACHE_DIR}")
+logger.info(f"History file: {HISTORY_FILE}")
+
+# Initialize empty history file if it doesn't exist
+if not os.path.exists(HISTORY_FILE):
+    with open(HISTORY_FILE, 'w') as f:
+        json.dump({'history': [], 'metrics': {
+            'total_reviews': 0,
+            'avg_response_time': 0.0,
+            'reviews_today': 0
+        }}, f)
+    # Ensure history file has write permissions
+    os.chmod(HISTORY_FILE, stat.S_IRUSR | stat.S_IWUSR | stat.S_IRGRP | stat.S_IWGRP | stat.S_IROTH | stat.S_IWOTH)
+    logger.info("Created empty history file")
 
 class Review:
     def __init__(self, code: str, language: str, suggestions: str):
@@ -79,6 +99,13 @@ class CodeReviewer:
                 logger.info(f"Loaded {len(self.review_history)} reviews from history")
         except Exception as e:
             logger.error(f"Error loading history: {e}")
+            # Initialize empty history if file doesn't exist or is corrupted
+            self.review_history = []
+            self.metrics = {
+                'total_reviews': 0,
+                'avg_response_time': 0.0,
+                'reviews_today': 0
+            }
     
     def save_history(self):
         """Save review history to file."""
@@ -87,8 +114,14 @@ class CodeReviewer:
                 'history': [r.to_dict() for r in self.review_history],
                 'metrics': self.metrics
             }
+            # Ensure directory exists with proper permissions
+            os.makedirs(os.path.dirname(HISTORY_FILE), exist_ok=True)
+            os.chmod(os.path.dirname(HISTORY_FILE), stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
+            
             with open(HISTORY_FILE, 'w') as f:
                 json.dump(data, f)
+            # Ensure file has proper permissions
+            os.chmod(HISTORY_FILE, stat.S_IRUSR | stat.S_IWUSR | stat.S_IRGRP | stat.S_IWGRP | stat.S_IROTH | stat.S_IWOTH)
             logger.info("Saved review history")
         except Exception as e:
             logger.error(f"Error saving history: {e}")
@@ -104,7 +137,7 @@ class CodeReviewer:
         """Initialize the model and tokenizer."""
         try:
             if HF_TOKEN:
-                login(token=HF_TOKEN, add_to_git_credential=False)
+                login(token=HF_TOKEN, add_to_git_credential=True)
             
             logger.info("Loading tokenizer...")
             self.tokenizer = AutoTokenizer.from_pretrained(
@@ -410,9 +443,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        show_error=True,
-        quiet=False
-    )
+    iface.launch()