Update app.py

app.py

@@ -13,6 +13,9 @@ if not openai_api_key:
 
 client = openai.OpenAI(api_key=openai_api_key)
 
+openrouter_key = os.environ.get("OPENROUTER")
+openrouter = openai.OpenAI(api_key=openrouter_key, base_url="https://openrouter.ai/api/v1")
+
 # --- Exadata Specs ---
 exadata_specs = {
     "X7": {"Quarter Rack": {"max_iops": 350000, "max_throughput": 25}, "Half Rack": {"max_iops": 700000, "max_throughput": 50}, "Full Rack": {"max_iops": 1400000, "max_throughput": 100}},
@@ -128,7 +131,7 @@ AWR CONTENT:
 """
 
         response = client.chat.completions.create(
-            model="gpt-4-turbo",
+            model=llm_model,
             messages=[
                 {"role": "system", "content": "You are the strict Oracle AWR Health Analysis Agent."},
                 {"role": "user", "content": prompt}
@@ -137,16 +140,18 @@ AWR CONTENT:
 
         return response.choices[0].message.content.strip()
 
-
 class RaterAgent:
-    def rate(self, content, llm_model):
+    def rate(self, content):
         prompt = f"Rate the following analysis from 1-5 stars and explain:\n\n{content}"
-        response = client.chat.completions.create(
-            model=llm_model,
+        
+        response = openrouter.chat.completions.create(
+            model="mistralai/Mixtral-8x7B-Instruct",  # FORCE to use Mixtral via OpenRouter
             messages=[{"role": "user", "content": prompt}]
         )
+        
         return response.choices[0].message.content.strip()
 
+
 # === Main Process ===
 def process_awr(awr_text, threshold, performance_test_mode, exadata_model, rack_size, llm_model):
     analyzer = CriticalAnalyzerAgent()
@@ -158,7 +163,7 @@ def process_awr(awr_text, threshold, performance_test_mode, exadata_model, rack_
 
     analysis = analyzer.analyze(awr_text, performance_test_mode, exadata_model, rack_size, llm_model)
     health_status = health.check_health(awr_text, llm_model)
-    rating_text = rater.rate(analysis, llm_model)
+    rating_text = rater.rate(analysis)
 
     stars = 0
     match = re.search(r"(\d+)", rating_text)