Update config.py

config.py

@@ -1,29 +1,32 @@
-# config.py
-
 import os
 from dotenv import load_dotenv
 from typing import Dict, Any
 
+# Load environment variables from .env file
 load_dotenv()
-GROQ_API_KEY="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"
+
+# Groq API Key (prefer loading from environment variables for security)
+GROQ_API_KEY = os.getenv("GROQ_API_KEY", "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")
 
 # AI Provider Configuration
 AI_PROVIDER = "groq"  # We're now using Groq exclusively
 AI_MODEL = os.getenv("AI_MODEL", "deepseek-ai/deepseek-r1-distill-llama-70b")  # Default model
-AI_API_KEY = GROQ_API_KEY
+AI_API_KEY = GROQ_API_KEY  # Use the Groq API key
 
-# Model Settings
+# Model Settings (Groq-specific settings)
 MODEL_SETTINGS = {
-    "max_tokens": 4000,
-    "temperature": 0.7,
-    "top_p": 1.0
+    "max_tokens": 4000,  # Maximum tokens for the response
+    "temperature": 0.7,  # Controls randomness (0.0 = deterministic, 1.0 = creative)
+    "top_p": 1.0,  # Controls diversity via nucleus sampling
+    "frequency_penalty": 0.0,  # Penalizes repeated tokens
+    "presence_penalty": 0.0,  # Penalizes new tokens
 }
 
 # Language Settings
 DEFAULT_LANGUAGE = 'ar'  # Arabic by default
-SUPPORTED_LANGUAGES = ['ar', 'en']
+SUPPORTED_LANGUAGES = ['ar', 'en']  # Supported languages for the application
 
-# UAE Legal Resources
+# UAE Legal Resources (Domains for legal research)
 UAE_LEGAL_DOMAINS = [
     'https://elaws.moj.gov.ae',
     'https://www.mohre.gov.ae',
@@ -32,7 +35,7 @@ UAE_LEGAL_DOMAINS = [
     'https://www.dc.gov.ae'
 ]
 
-# Legal Categories
+# Legal Categories (Mapping of legal categories to their Arabic names)
 LEGAL_CATEGORIES = {
     'civil': 'القانون المدني',
     'criminal': 'القانون الجنائي',
@@ -43,12 +46,13 @@ LEGAL_CATEGORIES = {
 }
 
 def get_ai_config() -> Dict[str, Any]:
-    """Get AI configuration based on environment variables"""
+    """
+    Get AI configuration based on environment variables.
+    Returns a dictionary containing the provider, model, API key, and model settings.
+    """
     return {
         "provider": AI_PROVIDER,
         "model": AI_MODEL,
         "api_key": AI_API_KEY,
-        "temperature": MODEL_SETTINGS["temperature"],
-        "max_tokens": MODEL_SETTINGS["max_tokens"],
-        "top_p": MODEL_SETTINGS["top_p"]
+        **MODEL_SETTINGS  # Include all model settings in the config
     }