Fix model name and storage paths for metrics/history

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
-import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
 from huggingface_hub import login
 import os
 import logging
@@ -19,22 +19,16 @@ logger = logging.getLogger(__name__)
 
 # Environment variables
 HF_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
-MODEL_NAME = os.getenv("MODEL_NAME", "google/gemma-2-2b-it")
-
-# Login to Hugging Face with git credential
-if HF_TOKEN:
-    try:
-        login(token=HF_TOKEN, add_to_git_credential=True)
-        logger.info("Successfully logged in to Hugging Face")
-    except Exception as e:
-        logger.error(f"Error logging in to Hugging Face: {e}")
+MODEL_NAME = os.getenv("MODEL_NAME", "google/gemma-2-2b-it")  # Fixed model name
 
 # Create data directory for persistence
 DATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "data")
 os.makedirs(DATA_DIR, exist_ok=True)
 
-# History file
+# 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)
 
 class Review:
     def __init__(self, code: str, language: str, suggestions: str):
@@ -85,13 +79,6 @@ 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."""
@@ -100,8 +87,6 @@ class CodeReviewer:
                 'history': [r.to_dict() for r in self.review_history],
                 'metrics': self.metrics
             }
-            # Ensure the directory exists
-            os.makedirs(os.path.dirname(HISTORY_FILE), exist_ok=True)
             with open(HISTORY_FILE, 'w') as f:
                 json.dump(data, f)
             logger.info("Saved review history")
@@ -118,11 +103,15 @@ class CodeReviewer:
     def initialize_model(self):
         """Initialize the model and tokenizer."""
         try:
+            if HF_TOKEN:
+                login(token=HF_TOKEN, add_to_git_credential=False)
+            
             logger.info("Loading tokenizer...")
             self.tokenizer = AutoTokenizer.from_pretrained(
                 MODEL_NAME,
                 token=HF_TOKEN,
-                trust_remote_code=True
+                trust_remote_code=True,
+                cache_dir=CACHE_DIR
             )
             special_tokens = {
                 'pad_token': '[PAD]',
@@ -140,6 +129,7 @@ class CodeReviewer:
                 torch_dtype=torch.float16,
                 trust_remote_code=True,
                 low_cpu_mem_usage=True,
+                cache_dir=CACHE_DIR,
                 token=HF_TOKEN
             )
             if num_added > 0:
@@ -418,5 +408,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         inputs=[code_input, language_input]
     )
 
+# Launch the app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True,
+        quiet=False
+    )