Update app.py

app.py

@@ -1,17 +1,301 @@
+import gradio as gr
+import requests
+import os
+import base64
+import io
+import datetime
+from PIL import Image
+
+# =========================================================
+# ENV CONFIG
+# =========================================================
+
+OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
+BREVO_API_KEY = os.getenv("BREVO_API_KEY")
+BREVO_FROM_EMAIL = os.getenv("BREVO_FROM_EMAIL", "noreply@company.com")
+DEFAULT_REPORT_EMAIL = os.getenv("REPORT_EMAIL", "")
+
+conversation_history = []
+current_report_email = DEFAULT_REPORT_EMAIL
+
+# =========================================================
+# DEPARTMENTS
+# =========================================================
+
+departments = {
+    "administration": {
+        "name": "Administration",
+        "email": os.getenv("ADMIN_EMAIL", "")
+    },
+    "production": {
+        "name": "Production",
+        "email": os.getenv("PRODUCTION_EMAIL", "")
+    },
+    "magazine": {
+        "name": "Magazine Management",
+        "email": os.getenv("MAGAZINE_EMAIL", "")
+    }
+}
+
+# =========================================================
+# EMAIL (BREVO)
+# =========================================================
+
+def send_email_brevo_api(to_email, subject, html):
+    if not BREVO_API_KEY:
+        return "❌ Brevo API key missing"
+
+    r = requests.post(
+        "https://api.brevo.com/v3/smtp/email",
+        headers={
+            "accept": "application/json",
+            "content-type": "application/json",
+            "api-key": BREVO_API_KEY
+        },
+        json={
+            "sender": {
+                "name": "AI Coordination System",
+                "email": BREVO_FROM_EMAIL
+            },
+            "to": [{"email": to_email}],
+            "subject": subject,
+            "htmlContent": html
+        },
+        timeout=20
+    )
+
+    return "✅ Email sent" if r.status_code == 201 else f"❌ {r.text}"
+
+
+# =========================================================
+# LLM ANALYSIS (PROFESSIONAL VERSION)
+# =========================================================
+
+SYSTEM_PROMPT = """
+You are an Enterprise Operations Coordination Assistant.
+
+Your role:
+- Analyze inter-department requests
+- Write formally and professionally
+- Be concise, structured, executive-style
+- No emojis
+- No casual language
+- Provide actionable steps only
+
+Always follow the exact output structure requested.
+"""
+
+def call_llm(messages):
+    r = requests.post(
+        "https://openrouter.ai/api/v1/chat/completions",
+        headers={
+            "Authorization": f"Bearer {OPENROUTER_API_KEY}",
+            "Content-Type": "application/json",
+        },
+        json={
+            "model": "google/gemma-3-27b-it:free",
+            "messages": messages,
+            "temperature": 0.1,
+            "max_tokens": 800
+        },
+        timeout=60
+    )
+
+    if r.status_code == 200:
+        return r.json()["choices"][0]["message"]["content"]
+
+    return "LLM analysis failed."
+
+
+def analyze_coordination_needs(message, department, user_name, image=None):
+
+    if image:
+        buffer = io.BytesIO()
+        image.save(buffer, format="PNG")
+        img64 = base64.b64encode(buffer.getvalue()).decode()
+
+        user_prompt = f"""
+Analyze the following request and attached image.
+
+Department: {department}
+Requestor: {user_name}
+Message: {message}
+
+Return:
+
+=== SUMMARY ===
+Brief executive summary
+
+=== IMAGE OBSERVATIONS ===
+Relevant findings
+
+=== DEPARTMENTS INVOLVED ===
+Bullets
+
+=== ACTION PLAN ===
+Numbered steps
+
+=== PRIORITY ===
+Low | Medium | High with reason
+
+=== FOLLOW-UP ===
+Next checkpoints
+"""
+
+        return call_llm([
+            {"role": "system", "content": SYSTEM_PROMPT},
+            {
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": user_prompt},
+                    {"type": "image_url", "image_url": {"url": f"data:image/png;base64,{img64}"}}
+                ]
+            }
+        ])
+
+    else:
+
+        user_prompt = f"""
+Analyze this inter-department request.
+
+Department: {department}
+Requestor: {user_name}
+Message: {message}
+
+Return:
+
 === COORDINATION SUMMARY ===
-Production line downtime reported requiring immediate technical intervention.
+Short summary
 
 === DEPARTMENTS INVOLVED ===
-• Production – technical inspection
-• Administration – approval of maintenance budget
+Bullets
 
 === RECOMMENDED ACTION PLAN ===
-1. Inspect equipment
-2. Log maintenance ticket
-3. Schedule repair within 24 hours
+Numbered steps
 
 === PRIORITY LEVEL ===
-High – Direct operational impact
+Low | Medium | High with justification
 
 === FOLLOW-UP ===
-Daily status update until resolution
+Monitoring or reporting
+"""
+
+        return call_llm([
+            {"role": "system", "content": SYSTEM_PROMPT},
+            {"role": "user", "content": user_prompt}
+        ])
+
+
+# =========================================================
+# HISTORY
+# =========================================================
+
+def add_history(entry):
+    conversation_history.append(entry)
+    if len(conversation_history) > 50:
+        conversation_history.pop(0)
+
+
+# =========================================================
+# MAIN PROCESS
+# =========================================================
+
+def process_request(name, department, message, image, recipient, send_email, urgency):
+
+    if not name or not message or not department:
+        return "Please complete all required fields."
+
+    analysis = analyze_coordination_needs(message, department, name, image)
+
+    response = f"""
+Requestor: {name}
+Department: {departments[department]['name']}
+
+{analysis}
+"""
+
+    email_status = ""
+
+    if send_email and recipient:
+        recipient_email = departments[recipient]["email"] or current_report_email
+
+        html = f"""
+        <pre style="font-family:Arial">{response}</pre>
+        """
+
+        email_status = send_email_brevo_api(
+            recipient_email,
+            f"[{urgency}] Coordination Request",
+            html
+        )
+
+        response += f"\n\nEmail Status: {email_status}"
+
+    add_history({
+        "time": datetime.datetime.now().isoformat(),
+        "user": name,
+        "department": department,
+        "message": message,
+        "analysis": analysis
+    })
+
+    return response
+
+
+# =========================================================
+# STATS
+# =========================================================
+
+def stats():
+    return f"Total interactions: {len(conversation_history)}"
+
+
+def clear():
+    conversation_history.clear()
+    return "History cleared."
+
+
+# =========================================================
+# UI
+# =========================================================
+
+with gr.Blocks(theme=gr.themes.Soft(), title="AI Coordination System") as ui:
+
+    gr.Markdown("# 🏢 AI Department Coordination System")
+
+    with gr.Row():
+
+        with gr.Column(scale=1):
+
+            name = gr.Textbox(label="Name")
+            dept = gr.Dropdown(list(departments.keys()), label="Department")
+            message = gr.Textbox(lines=4, label="Request")
+            image = gr.Image(type="pil", label="Optional Image")
+
+            recipient = gr.Dropdown(list(departments.keys()), label="Notify Department")
+            urgency = gr.Dropdown(["Low", "Medium", "High"], value="Medium")
+
+            send_email = gr.Checkbox(label="Send Email", value=True)
+
+            submit = gr.Button("Analyze", variant="primary")
+            stats_btn = gr.Button("Stats")
+            clear_btn = gr.Button("Clear")
+
+        with gr.Column(scale=2):
+            output = gr.Textbox(lines=20, show_copy_button=True)
+
+    submit.click(
+        process_request,
+        inputs=[name, dept, message, image, recipient, send_email, urgency],
+        outputs=output
+    )
+
+    stats_btn.click(stats, outputs=output)
+    clear_btn.click(clear, outputs=output)
+
+
+# =========================================================
+
+if __name__ == "__main__":
+    ui.launch()
+