Update app.py

app.py

@@ -1,3 +1,4 @@
+
 # === Imports ===
 import os
 import re
@@ -17,16 +18,7 @@ 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")
 
-# --- OCI Object Storage Config ---
-#oci_config = {
-#    "user": os.environ.get("OCI_USER"),
-#    "tenancy": os.environ.get("OCI_TENANCY"),
-#    "fingerprint": os.environ.get("OCI_FINGERPRINT"),
-#    "region": os.environ.get("OCI_REGION"),
-#    "key_content": os.environ.get("OCI_PRIVATE_KEY"),
-#}
-
-# === OCI Object Storage Setup ===
+# --- OCI Object Storage Setup ---
 oci_config = {
     "user": os.environ.get("OCI_USER"),
     "tenancy": os.environ.get("OCI_TENANCY"),
@@ -43,11 +35,6 @@ try:
 except Exception as e:
     print("Failed to initialize OCI Object Storage client:", e)
 
-
-namespace = os.environ.get("OCI_NAMESPACE")
-bucket_name = os.environ.get("OCI_BUCKET_NAME")
-object_storage = oci.object_storage.ObjectStorageClient(oci_config)
-
 # --- 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}},
@@ -75,7 +62,7 @@ def upload_awr_file(file_obj):
     filename = os.path.basename(file_obj.name)
     content = file_obj.read()
     object_storage.put_object(namespace, bucket_name, filename, content)
-    return f"\u2705 Uploaded {filename}"
+    return f"✅ Uploaded {filename}"
 
 def list_awr_files():
     try:
@@ -99,27 +86,145 @@ def compare_awrs(file_list, llm_model):
     for fname in file_list:
         content = get_awr_file_text(fname)
         combined_text += f"\n=== AWR: {fname} ===\n{content[:3000]}...\n"
-    prompt = f"""
-You are a senior Oracle performance engineer. You will compare multiple AWR reports and highlight:
+    prompt = f"""You are a senior Oracle performance engineer. You will compare multiple AWR reports and highlight:
 - Key differences in workload or system behavior
 - Major trends or anomalies
 - Which report shows better performance and why
 - Exadata-specific metrics like Smart Scan, Flash I/O
 - Suggestions to unify or improve system behavior
+
 AWR Reports:
 {combined_text}
 """
     response = client.chat.completions.create(
         model=llm_model,
-        messages=[{"role": "system", "content": "You are a comparative AWR analysis expert."},
-                  {"role": "user", "content": prompt}]
+        messages=[
+            {"role": "system", "content": "You are a comparative AWR analysis expert."},
+            {"role": "user", "content": prompt}
+        ]
     )
     return response.choices[0].message.content.strip()
 
+# === AGENTS ===
+class CriticalAnalyzerAgent:
+    def analyze(self, content, performance_test_mode, exadata_model, rack_size, llm_model):
+        cleaned_content = clean_awr_content(content)
+        if len(cleaned_content) > 128000:
+            cleaned_content = cleaned_content[:128000] + "\n\n[TRUNCATED]..."
+
+        prompt = f"""You are an expert Oracle DBA performance analyst specialized in AWR + Exadata.
+
+Please perform advanced analysis on the following report:
+
+======== AWR REPORT START ========
+{cleaned_content}
+======== AWR REPORT END ========
+
+Required Output:
+- Performance Summary (with metric values)
+- Detailed Bottlenecks + Risks (quantified)
+- Forecast + Predictions
+- Monitoring Recommendations
+- Exadata Statistics (IO, Flash Cache, Smart Scan)
+- Recommended Next Steps to Bridge Gaps
+"""
+        if performance_test_mode and exadata_model and rack_size:
+            specs = exadata_specs.get(exadata_model, {}).get(rack_size, {})
+            if specs:
+                prompt += f"""
+
+This was a PERFORMANCE TEST on Oracle Exadata {exadata_model} {rack_size}.
+Theoretical Max:
+- IOPS: {specs['max_iops']}
+- Throughput: {specs['max_throughput']} GB/s
+
+Compare observed vs theoretical. Recommend actions to close the performance gap.
+"""
+        response = client.chat.completions.create(
+            model=llm_model,
+            messages=[
+                {"role": "system", "content": "You are an expert Oracle DBA."},
+                {"role": "user", "content": prompt}
+            ]
+        )
+        return response.choices[0].message.content.strip()
+
+class HealthAgent:
+    def check_health(self, content, llm_model):
+        cleaned_content = clean_awr_content(content)
+        if len(cleaned_content) > 128000:
+            cleaned_content = cleaned_content[:128000] + "\n\n[TRUNCATED]..."
+
+        prompt = f"""You are the Oracle AWR Health Analysis Agent.
+
+Your primary responsibility is to detect and report ANY and ALL database health risks, alerts, warnings, or failures in the AWR report.
+
+You MUST:
+- Identify all issues marked as CRITICAL, WARNING, ALERT, FAILED, OFFLINE, CONFINED, DROPPED, or ERROR.
+- Never omit or generalize. If something appears important, call it out.
+- Classify each issue into: 🚨 CRITICAL / ⚠️ WARNING / ✅ INFO
+- For CRITICAL and WARNING, provide suggested actions or considerations.
+- Always confirm at the end if no CRITICAL or WARNING issues were found.
+
+Special Attention Areas:
+- Flash Cache or Flash Disk Failures
+- I/O Subsystem stalls or errors
+- ASM/Grid Disk issues
+- Smart Scan failures
+- Redo Log issues
+- RAC Interconnect issues
+
+AWR CONTENT:
+{cleaned_content}
+"""
+        response = client.chat.completions.create(
+            model=llm_model,
+            messages=[
+                {"role": "system", "content": "You are the strict Oracle AWR Health Analysis Agent."},
+                {"role": "user", "content": prompt}
+            ]
+        )
+        return response.choices[0].message.content.strip()
+
+class RaterAgent:
+    def rate(self, content):
+        prompt = f"Rate the following analysis from 1-5 stars and explain:\n\n{content}"
+        response = openrouter.chat.completions.create(
+            model="mistralai/Mixtral-8x7B-Instruct",
+            messages=[{"role": "user", "content": prompt}]
+        )
+        return response.choices[0].message.content.strip()
+
+# === MAIN AWR PROCESS ===
+def process_awr(awr_text, threshold, performance_test_mode, exadata_model, rack_size, llm_model):
+    analyzer = CriticalAnalyzerAgent()
+    health = HealthAgent()
+    rater = RaterAgent()
+
+    if not awr_text.strip():
+        return "No AWR text provided", "", "", ""
+
+    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)
+
+    stars = 0
+    match = re.search(r"(\d+)", rating_text)
+    if match:
+        stars = int(match.group(1))
+
+    retry_status = "✅ Accepted"
+    if stars < threshold:
+        analysis = analyzer.analyze(awr_text, performance_test_mode, exadata_model, rack_size, llm_model)
+        rating_text = rater.rate(analysis)
+        retry_status = "✅ Retry Occurred"
+
+    return analysis, health_status, rating_text, retry_status
+
 # === Gradio UI ===
 with gr.Blocks() as demo:
     with gr.Tab("Manual AWR Analysis"):
-        gr.Markdown("# \U0001f9e0 Multi-Agent Oracle AWR Analyzer (Production Edition)")
+        gr.Markdown("# 🧠 Multi-Agent Oracle AWR Analyzer (Production Edition)")
         awr_text = gr.Textbox(label="Paste AWR Report", lines=30)
         threshold = gr.Slider(0, 5, value=3, step=1, label="Correctness Threshold (Stars)")
         performance_test_mode = gr.Checkbox(label="Performance Test Mode")
@@ -137,7 +242,6 @@ with gr.Blocks() as demo:
         rating = gr.Textbox(label="Rater", lines=3)
         retry_status = gr.Textbox(label="Retry Status")
 
-        from your_existing_code import process_awr  # Replace with actual import or include function here
         analyze_btn.click(process_awr,
                           inputs=[awr_text, threshold, performance_test_mode, exadata_model, rack_size, llm_selector],
                           outputs=[output, health, rating, retry_status])
@@ -147,7 +251,7 @@ with gr.Blocks() as demo:
         upload_status = gr.Textbox(label="Upload Status")
         upload_file.upload(fn=upload_awr_file, inputs=upload_file, outputs=upload_status)
 
-        refresh_button = gr.Button("\U0001f503 Refresh File List")
+        refresh_button = gr.Button("🔃 Refresh File List")
         file_multiselect = gr.Dropdown(choices=[], label="Select AWR Files", multiselect=True)
         refresh_button.click(fn=lambda: gr.update(choices=list_awr_files()), outputs=file_multiselect)
 
@@ -157,4 +261,5 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     demo.launch(debug=True)
-    #
+
+