app.py

import os
import uuid
from datetime import datetime, timezone

import gradio as gr
from azure.cosmos import CosmosClient, PartitionKey
from azure.cosmos.exceptions import CosmosHttpResponseError
from dotenv import load_dotenv

load_dotenv()


def _get_env(name: str) -> str:
    value = os.getenv(name, "").strip()
    return value


def get_container():
    endpoint = _get_env("COSMOS_ENDPOINT")
    key = _get_env("COSMOS_KEY")
    db_name = _get_env("COSMOS_DATABASE")
    container_name = _get_env("COSMOS_CONTAINER")

    if not all([endpoint, key, db_name, container_name]):
        missing = [
            k
            for k in [
                "COSMOS_ENDPOINT",
                "COSMOS_KEY",
                "COSMOS_DATABASE",
                "COSMOS_CONTAINER",
            ]
            if not _get_env(k)
        ]
        raise ValueError(
            "Missing Cosmos configuration: " + ", ".join(missing)
        )

    client = CosmosClient(endpoint, credential=key)
    database = client.create_database_if_not_exists(id=db_name)
    container = database.create_container_if_not_exists(
        id=container_name,
        partition_key=PartitionKey(path="/id"),
    )
    return container


_container_cache = None


def submit_feedback(name, team, assistant_type, submit_anonymous, do_text, dont_text):
    name = (name or "").strip()
    team = (team or "").strip()
    assistant_type = (assistant_type or "").strip()
    do_text = (do_text or "").strip()
    dont_text = (dont_text or "").strip()

    if not do_text and not dont_text:
        return (
            (
                "<div style=\"padding:14px 16px;border-radius:8px;"
                "background:#ffe9e9;color:#7a1d1d;border:1px solid #f2bcbc;"
                "font-weight:600;\">Please enter a Dos or a Don'ts (or both) before submitting.</div>"
            ),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
        )

    if submit_anonymous:
        name = ""
        team = ""

    global _container_cache
    try:
        if _container_cache is None:
            _container_cache = get_container()

        item = {
            "id": str(uuid.uuid4()),
            "created_at": datetime.now(timezone.utc).isoformat(),
            "name": name or None,
            "team": team or None,
            "assistant_type": assistant_type or None,
            "do": do_text or None,
            "dont": dont_text or None,
        }
        _container_cache.create_item(body=item)
    except ValueError as exc:
        return (
            (
                "<div style=\"padding:14px 16px;border-radius:8px;"
                "background:#ffe9e9;color:#7a1d1d;border:1px solid #f2bcbc;"
                f"font-weight:600;\">Configuration error: {exc}</div>"
            ),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
        )
    except CosmosHttpResponseError as exc:
        return (
            (
                "<div style=\"padding:14px 16px;border-radius:8px;"
                "background:#ffe9e9;color:#7a1d1d;border:1px solid #f2bcbc;"
                f"font-weight:600;\">Cosmos DB error: {exc.message}</div>"
            ),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
        )
    except Exception as exc:  # noqa: BLE001
        return (
            (
                "<div style=\"padding:14px 16px;border-radius:8px;"
                "background:#ffe9e9;color:#7a1d1d;border:1px solid #f2bcbc;"
                f"font-weight:600;\">Unexpected error: {exc}</div>"
            ),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
            gr.update(),
        )

    return (
        (
            "<div style=\"padding:16px 18px;border-radius:10px;"
            "background:#e8f7ee;color:#0f5b2f;border:1px solid #bfe5cc;"
            "font-weight:700;font-size:16px;\">✅ Saved! Thanks for your feedback.</div>"
        ),
        gr.update(value=""),
        gr.update(value=""),
        gr.update(value="General"),
        gr.update(value=False),
        gr.update(value=""),
        gr.update(value=""),
    )


with gr.Blocks(title="AI Assistant Do/Don't Collection") as demo:
    gr.Markdown(
        "# AI Assistant Do/Don't Collection\n"
        "Share guidance for building our AI assistants. You can fill either field or both."
    )

    with gr.Row():
        name_input = gr.Textbox(label="Your name (optional)", placeholder="Jane Doe")
        team_input = gr.Textbox(label="Team / Org (optional)", placeholder="Product, IT, ...")
        assistant_type_input = gr.Dropdown(
            label="Assistant type (optional)",
            choices=[
                "General",
                "Resiliency Programs (WFPH, FORTIFIED, ..)",
                "Communication",
                "Code and Standards",
                "Internal Ops",
                "Research",
            ],
            value="General",
        )

    submit_anonymous_input = gr.Checkbox(
        label="Submit anonymously (ignore name and team)",
        value=False,
    )

    do_input = gr.Textbox(
        label="Dos",
        placeholder="What should we consider when creating AI assistants?",
        lines=6,
    )
    dont_input = gr.Textbox(
        label="Don'ts",
        placeholder="What should we avoid when creating AI assistants?",
        lines=6,
    )

    submit_btn = gr.Button("Submit")
    status = gr.HTML()

    submit_btn.click(
        fn=submit_feedback,
        inputs=[
            name_input,
            team_input,
            assistant_type_input,
            submit_anonymous_input,
            do_input,
            dont_input,
        ],
        outputs=[
            status,
            name_input,
            team_input,
            assistant_type_input,
            submit_anonymous_input,
            do_input,
            dont_input,
        ],
    )


if __name__ == "__main__":
    demo.launch()