Update app.py

app.py

@@ -1,124 +1,100 @@
-# app.py
 import os
-import datetime
-import json
 from pathlib import Path
-from reportlab.lib.pagesizes import letter
-from reportlab.pdfgen import canvas
 import gradio as gr
 
-# -----------------------
-# CONFIG
-# -----------------------
-FILE_STORAGE = Path("file_storage")
-FILE_STORAGE.mkdir(exist_ok=True)
-REPORT_DIR = Path("reports")
-REPORT_DIR.mkdir(exist_ok=True)
-
-# -----------------------
-# LOG ANALYSIS FUNCTIONS
-# -----------------------
-def parse_log_text(txt):
-    lines = txt.splitlines()
-    errors = [l for l in lines if "ERROR" in l or "CRITICAL" in l or "FATAL" in l]
-    warnings = [l for l in lines if "WARNING" in l or "WARN" in l]
-    return {"errors": errors, "warnings": warnings, "total_lines": len(lines)}
-
-def analyze_log(filename):
-    filepath = FILE_STORAGE / filename
-    if not filepath.exists():
-        return {"summary":"File not found","issues":[],"suggestions":[],"plan":""}
-    
-    text = filepath.read_text(encoding="utf-8", errors="ignore")
-    parsed = parse_log_text(text)
-    
-    summary = f"Detected {len(parsed['errors'])} errors and {len(parsed['warnings'])} warnings in {parsed['total_lines']} lines."
-    issues = parsed['errors'][:10] + parsed['warnings'][:10]
-    suggestions = [
-        "Investigate top errors",
-        "Check system configuration",
-        "Add proper exception handling"
-    ]
-    plan = "\n".join([
-        "STEP 1: Review errors",
-        "STEP 2: Apply suggested fixes",
-        "STEP 3: Re-run analysis"
-    ])
-    return {"summary": summary, "issues": issues, "suggestions": suggestions, "plan": plan}
-
-def generate_report(filename, summary, issues, suggestions):
-    now = datetime.datetime.utcnow().strftime("%Y%m%d-%H%M%S")
-    out = REPORT_DIR / f"incident_{filename}_{now}.pdf"
-    c = canvas.Canvas(str(out), pagesize=letter)
-    c.setFont("Helvetica-Bold",16)
-    c.drawString(40,750,f"Incident Report - {filename}")
-    c.setFont("Helvetica",10)
-    c.drawString(40,730,f"Generated: {datetime.datetime.utcnow().isoformat()}")
-    y=700
-    c.drawString(40,y,"Summary:")
-    c.drawString(60,y-20,summary[:1000])
-    y-=60
-    c.drawString(40,y,"Issues:")
-    for it in issues:
-        y-=14
-        c.drawString(60,y,it[:120])
-        if y<80:
-            c.showPage()
-            y=740
-    y-=20
-    c.drawString(40,y,"Suggestions:")
-    for s in suggestions:
-        y-=14
-        c.drawString(60,y,f"- {s[:100]}")
-        if y<80:
-            c.showPage()
-            y=740
-    c.save()
-    return str(out)
-
-# -----------------------
-# GRADIO UI
-# -----------------------
-def refresh_files():
-    return [f.name for f in FILE_STORAGE.iterdir() if f.is_file()]
-
-def upload_file(file_obj):
-    if file_obj is None:
-        return refresh_files()
-    filepath = FILE_STORAGE / file_obj.name
-    content = file_obj.read()
-    if isinstance(content, bytes):
-        content = content.decode("utf-8", errors="ignore")
-    filepath.write_text(content, encoding="utf-8")
-    return refresh_files()
-
-def analyze_selected(filename):
+# ------------------------------------------------------
+# STORAGE
+# ------------------------------------------------------
+STORAGE = Path("file_storage")
+STORAGE.mkdir(exist_ok=True)
+
+# ------------------------------------------------------
+# PARSER
+# ------------------------------------------------------
+def parse_logs(text):
+    lines = text.splitlines()
+
+    errors = [l for l in lines if "ERROR" in l.upper() or "CRITICAL" in l.upper()]
+    warnings = [l for l in lines if "WARNING" in l.upper()]
+
+    summary = f"Lines: {len(lines)} | Errors: {len(errors)} | Warnings: {len(warnings)}"
+
+    return summary, "\n".join(errors), "\n".join(warnings)
+
+
+# ------------------------------------------------------
+# UPLOAD HANDLER
+# ------------------------------------------------------
+def upload_file(file):
+    if file is None:
+        return "No file uploaded.", []
+
+    text = file.read().decode("utf-8", errors="ignore")
+
+    # Save it
+    path = STORAGE / file.name
+    with open(path, "w", encoding="utf-8") as f:
+        f.write(text)
+
+    # Update dropdown list
+    file_list = [p.name for p in STORAGE.iterdir()]
+
+    return f"Uploaded: {file.name}", file_list
+
+
+# ------------------------------------------------------
+# ANALYZE HANDLER
+# ------------------------------------------------------
+def analyze_file(filename):
     if not filename:
-        return "", "", "", "", None
-    out = analyze_log(filename)
-    report_path = generate_report(filename, out["summary"], out["issues"], out["suggestions"])
-    return (out["summary"], "\n".join(out["issues"]), "\n".join(out["suggestions"]), out["plan"], gr.File.update(value=report_path))
+        return "No file selected.", "", ""
+
+    path = STORAGE / filename
+    if not path.exists():
+        return "File not found.", "", ""
+
+    text = path.read_text(encoding="utf-8", errors="ignore")
+
+    summary, errors, warnings = parse_logs(text)
+
+    # ALWAYS return text; never empty components
+    return summary, errors if errors else "No errors found.", warnings if warnings else "No warnings found."
+
+
+# ------------------------------------------------------
+# BUILD UI
+# ------------------------------------------------------
+with gr.Blocks(theme="soft") as demo:
+
+    gr.Markdown("## **AgentOps MCP — Log Analyzer (Stable Version)**")
 
-with gr.Blocks() as demo:
-    gr.Markdown("## AgentOps MCP - Log Analysis")
     with gr.Row():
+        with gr.Column(scale=1):
+            upload = gr.File(label="Upload TXT Log File")
+            upload_out = gr.Textbox(label="Upload Status")
+
+            dropdown = gr.Dropdown(choices=[], label="Select File to Analyze")
+
+            btn_refresh = gr.Button("Refresh File List")
+            btn_analyze = gr.Button("Analyze File")
+
         with gr.Column(scale=2):
-            upload = gr.File(label="Upload log file", file_types=[".txt"])
-            btn_upload = gr.Button("Upload File")
-            files_dropdown = gr.Dropdown(choices=refresh_files(), label="Files available")
-            btn_refresh = gr.Button("Refresh files")
-            btn_analyze = gr.Button("Analyze selected file")
-        with gr.Column(scale=3):
             summary_box = gr.Textbox(label="Summary", lines=3)
-            issues_box = gr.Textbox(label="Detected Issues", lines=10)
-            suggestions_box = gr.Textbox(label="Suggestions", lines=6)
-            plan_box = gr.Textbox(label="Agent Plan", lines=6)
-            report_download = gr.File(label="Download PDF Report")
-
-    btn_refresh.click(fn=refresh_files, outputs=files_dropdown)
-    btn_upload.click(fn=upload_file, inputs=upload, outputs=files_dropdown)
-    btn_analyze.click(fn=analyze_selected, inputs=files_dropdown,
-                      outputs=[summary_box, issues_box, suggestions_box, plan_box, report_download])
-
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+            errors_box = gr.Textbox(label="Errors", lines=10)
+            warnings_box = gr.Textbox(label="Warnings", lines=10)
+
+    # ----- Events -----
+    upload.upload(fn=upload_file, inputs=upload, outputs=[upload_out, dropdown])
+
+    btn_refresh.click(
+        fn=lambda: [p.name for p in STORAGE.iterdir()],
+        outputs=dropdown
+    )
+
+    btn_analyze.click(
+        fn=analyze_file,
+        inputs=dropdown,
+        outputs=[summary_box, errors_box, warnings_box]
+    )
+
+demo.launch(server_name="0.0.0.0", server_port=7860)