Update app.py

app.py

@@ -1,24 +1,34 @@
 import gradio as gr
+from transformers import pipeline
 
-def devops_agent(query):
-    query = query.lower()
-    if "pod status" in query:
-        return "✅ All Kubernetes pods are running."
-    elif "restart jenkins" in query:
+# Load open-source LLM (distilled for speed)
+qa_model = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
+
+# Simulated DevOps context (can be replaced with real logs or docs)
+devops_context = """
+Kubernetes pods are running across three nodes. Jenkins is configured with two pipelines: build and deploy.
+Disk usage on node-1 is 72%. Recent logs show no errors. Deployment is triggered via GitHub Actions.
+"""
+
+def devops_qa_agent(query):
+    # Rule-based fallback
+    if "restart jenkins" in query.lower():
         return "🔄 Jenkins restarted successfully."
-    elif "deploy" in query:
+    elif "pod status" in query.lower():
+        return "✅ All Kubernetes pods are running."
+    elif "disk usage" in query.lower():
+        return "💽 Disk usage: 72% used on node-1."
+    elif "deploy" in query.lower():
         return "🚀 Deployment triggered via CI/CD pipeline."
-    elif "check logs" in query:
-        return "📜 Logs retrieved: No errors found."
-    elif "disk usage" in query:
-        return "💽 Disk usage: 72% used, 28% free."
     else:
-        return "🤖 Sorry, I didn’t understand that DevOps command."
+        # LLM-powered Q&A
+        result = qa_model(question=query, context=devops_context)
+        return f"🤖 LLM Answer: {result['answer']}"
 
 gr.Interface(
-    fn=devops_agent,
-    inputs=gr.Textbox(lines=2, placeholder="e.g. Check pod status"),
+    fn=devops_qa_agent,
+    inputs=gr.Textbox(lines=2, placeholder="Ask a DevOps question..."),
     outputs="text",
-    title="🛠️ DevOps AI Agent",
-    description="Enter a natural language DevOps command. The agent will simulate a response."
+    title="🛠️ DevOps AI Agent with Open-Source LLM",
+    description="Hybrid rule-based + LLM-powered Q&A agent. No API keys needed. Runs locally on Hugging Face Spaces."
 ).launch()