Create app.py

app.py

@@ -0,0 +1,149 @@
+import gradio as gr
+import re
+import cv2
+import numpy as np
+import pytesseract
+from PIL import Image, ImageDraw
+
+# Define keywords for highlighting
+KEYWORDS = ["fatal", "unreachable", "ssh", "publickey", "psrp", "5985", "python", "sudo", "8081",
+            "umask", "storage", "ssl/tls", "powershell", "authenticate", "hosts", "match", "playbook"]
+
+ERROR_TYPES = {
+    "ssh": "PORT ISSUE",
+    "publickey": "KEYS ISSUE",
+    "psrp": "PORT 5985 ISSUE",
+    "5985": "PORT 5985 ISSUE",
+    "python": "PYTHON ERROR",
+    "sudo": "SUDO ISSUE",
+    "8081": "PORT 8081 ISSUE",
+    "umask": "UMASK77 ISSUE",
+    "storage": "STORAGE ISSUE",
+    "ssl/tls": "SSL/TLS ISSUE",
+    "powershell": "POWERSHELL ISSUE",
+    "authenticate": "AUTHENTICATION ISSUE",
+    "hosts not match": "DEVICE NOT FOUND",
+    "playbook not": "PLAYBOOK NOT FOUND"
+}
+
+ERROR_DOCS = {
+    "KEYS ISSUE": "https://github.com/suryaT1/Documents/blob/main/keys-issue.md",
+    "PORT ISSUE": "https://github.com/suryaT1/Documents/blob/main/port-issue.md",
+    "PORT 5985 ISSUE": "https://github.com/suryaT1/Documents/blob/main/port5985-doc.md",
+    "PYTHON ERROR": "https://github.com/suryaT1/Documents/blob/main/python-issue.md",
+    "SUDO ISSUE": "https://github.com/suryaT1/Documents/blob/main/sudoers-issue.md",
+    "PORT 8081 ISSUE": "https://github.com/suryaT1/Documents/blob/main/port8081-issue.md",
+    "UMASK77 ISSUE": "https://github.com/suryaT1/Documents/blob/main/umask-issue.md",
+    "STORAGE ISSUE": "https://github.com/suryaT1/Documents/blob/main/storage-issue.md",
+    "SSL/TLS ISSUE": "https://github.com/suryaT1/Documents/blob/main/ssl_tls-issue.md",
+    "POWERSHELL ISSUE": "https://github.com/suryaT1/Documents/blob/main/powershell-issue.md",
+    "AUTHENTICATION ISSUE": "https://github.com/authentication-doc",
+    "DEVICE NOT FOUND": "https://github.com/suryaT1/Documents/blob/main/device-not-found.md",
+    "PLAYBOOK NOT FOUND": "https://github.com/suryaT1/Documents/blob/main/playbook-not-found.md"
+}
+
+DEVICE_NAME_PATTERN = r'\[([A-Za-z0-9@.\-]+)(?:\s*->\s*[A-Za-z0-9@.\-]+)?\]'
+
+# Highlight errors in text
+def highlight_keywords(text):
+    for keyword in KEYWORDS:
+        text = re.sub(rf'\b{keyword}\b', f'<b style="color:red;">{keyword}</b>', text, flags=re.IGNORECASE)
+    return text
+
+# Highlight errors directly on the image
+def highlight_errors_on_image(image_path):
+    image = Image.open(image_path)
+    image_cv = np.array(image)
+    image_cv = cv2.cvtColor(image_cv, cv2.COLOR_RGB2BGR)
+
+    data = pytesseract.image_to_data(image_cv, output_type=pytesseract.Output.DICT)
+
+    draw = ImageDraw.Draw(image)
+    for i, word in enumerate(data['text']):
+        if any(keyword.lower() in word.lower() for keyword in KEYWORDS):
+            x, y, w, h = data['left'][i], data['top'][i], data['width'][i], data['height'][i]
+            draw.rectangle([x, y, x + w, y + h], outline="red", width=3)
+            draw.text((x, y - 15), word, fill="red")
+
+    highlighted_image_path = "highlighted_image.png"
+    image.save(highlighted_image_path)
+    return highlighted_image_path, image
+
+# Analyze text input
+def analyze_text(text):
+    detected_errors = {}
+    devices = {}
+
+    device_matches = re.finditer(DEVICE_NAME_PATTERN, text)
+    for match in device_matches:
+        device = match.group(1)
+        device_text = text[match.end():].split("fatal:", 1)[0].lower()
+        found_ssh = "ssh" in device_text
+        found_publickey = "publickey" in device_text
+
+        if found_ssh and found_publickey:
+            detected_errors[device] = "KEYS ISSUE"
+        elif found_ssh:
+            detected_errors[device] = "PORT ISSUE"
+        elif "hosts" in device_text and "not" in device_text and "match" in device_text:
+            detected_errors[device] = "DEVICE NOT FOUND"
+        elif "playbook" in device_text and "not" in device_text:
+            detected_errors[device] = "PLAYBOOK NOT FOUND"
+        else:
+            for key, error_type in ERROR_TYPES.items():
+                if key in device_text:
+                    detected_errors[device] = error_type
+
+    return detected_errors
+
+# Process input
+def process_input(image, text):
+    if image is not None:
+        highlighted_image_path, highlighted_image = highlight_errors_on_image(image)
+        extracted_text = pytesseract.image_to_string(Image.open(image))
+        highlighted_text = ""
+    elif text.strip():
+        highlighted_image = ""
+        extracted_text = text.strip()
+        highlighted_text = highlight_keywords(extracted_text)
+    else:
+        return "", "No text detected!", ""
+
+    detected_errors = analyze_text(extracted_text)
+
+    logs_info = "<br>".join(f"Device: {device}, Issue: {error}" for device, error in detected_errors.items()) if detected_errors else "No errors detected"
+    docs_links = "<br>".join(f'<a href="{ERROR_DOCS[error]}" target="_blank">{error} Documentation</a>'
+                              for error in detected_errors.values() if error in ERROR_DOCS)
+
+    return highlighted_image if highlighted_image else highlighted_text, logs_info, docs_links
+
+# Show text input or image upload
+def show_input(choice):
+    return (gr.Textbox(visible=choice == "Text Box"),
+            gr.File(visible=choice == "Image Upload"))
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Ansible Log Annotator")
+    gr.Markdown("Select input type and analyze the logs.")
+
+    choice = gr.Radio(["Text Box", "Image Upload"], label="Select Input Type")
+    text_input = gr.Textbox(label="Enter Log Text", lines=5, visible=False)
+    image_input = gr.File(type="filepath", label="Upload Log Image", visible=False)
+    analyze_button = gr.Button("Analyze", visible=False)
+    output_display = gr.Image(label="Highlighted Output", visible=False)
+    output_logs = gr.HTML(label="Detected Issues", visible=False)
+    output_docs = gr.HTML(label="Documentation Links", visible=False)
+
+    choice.change(show_input, inputs=choice, outputs=[text_input, image_input])
+    choice.change(lambda: gr.Button(visible=True), outputs=[analyze_button])
+
+    analyze_button.click(process_input,
+                         inputs=[image_input, text_input],
+                         outputs=[output_display, output_logs, output_docs])
+
+    output_display.visible = True
+    output_logs.visible = True
+    output_docs.visible = True
+
+    demo.launch()