Update app.py

app.py

@@ -247,9 +247,36 @@ def process_awr(awr_text, threshold, performance_test_mode, exadata_model, rack_
 
 # === Gradio UI ===
 with gr.Blocks() as demo:
+
     with gr.Tab("Manual AWR Analysis"):
         gr.Markdown("# 🧠 Multi-Agent Oracle AWR Analyzer (Production Edition)")
-        awr_text = gr.Textbox(label="Paste AWR Report", lines=30)
+
+        # NEW: File upload for AWR
+        awr_file = gr.File(label="Upload AWR Report (.html or .txt)", file_types=[".html", ".txt"])
+        awr_text = gr.Textbox(label="AWR Report (pasted or loaded)", lines=30)
+
+        def awr_file_to_text(file_obj):
+            if not file_obj:
+                return ""
+            # Gradio File is a dict-like with 'name' and 'path'
+            filename = file_obj.name if hasattr(file_obj, "name") else str(file_obj)
+            # Support io.BytesIO for Gradio's in-memory files, fallback to path
+            try:
+                # Gradio may provide either path or file
+                content = file_obj.read() if hasattr(file_obj, "read") else open(file_obj, "rb").read()
+            except Exception:
+                with open(file_obj, "rb") as f:
+                    content = f.read()
+            # Decode as text (may need to try 'latin-1' for Oracle HTML)
+            try:
+                text = content.decode()
+            except Exception:
+                text = content.decode("latin-1")
+            return clean_awr_content(text)
+
+        awr_file.upload(awr_file_to_text, inputs=awr_file, outputs=awr_text)
+
+        # Everything below is the same as before
         threshold = gr.Slider(0, 5, value=3, step=1, label="Correctness Threshold (Stars)")
         performance_test_mode = gr.Checkbox(label="Performance Test Mode")
         exadata_model = gr.Dropdown(choices=list(exadata_specs.keys()), label="Exadata Model", visible=False)
@@ -266,9 +293,11 @@ with gr.Blocks() as demo:
         rating = gr.Textbox(label="Rater", lines=3)
         retry_status = gr.Textbox(label="Retry Status")
 
-        analyze_btn.click(process_awr,
-                          inputs=[awr_text, threshold, performance_test_mode, exadata_model, rack_size, llm_selector],
-                          outputs=[output, health, rating, retry_status])
+        analyze_btn.click(
+            process_awr,
+            inputs=[awr_text, threshold, performance_test_mode, exadata_model, rack_size, llm_selector],
+            outputs=[output, health, rating, retry_status]
+        )
 
     with gr.Tab("Compare AWRs from OCI"):
         upload_file = gr.File(label="Upload AWR Report", file_types=[".html", ".txt"])
@@ -281,9 +310,14 @@ with gr.Blocks() as demo:
 
         llm_compare = gr.Dropdown(choices=list(supported_llms.keys()), value="gpt-4-turbo", label="LLM Model for Comparison")
         compare_output = gr.Textbox(label="Comparison Output", lines=20)
-        gr.Button("Compare Selected AWRs").click(fn=compare_awrs, inputs=[file_multiselect, llm_compare], outputs=compare_output)
+        gr.Button("Compare Selected AWRs").click(
+            fn=compare_awrs,
+            inputs=[file_multiselect, llm_compare],
+            outputs=compare_output
+        )
 
 if __name__ == "__main__":
     demo.launch(debug=True)
 
+