Update app.py

app.py

@@ -1,252 +1,222 @@
 import os
-from typing import Tuple
-
 import gradio as gr
-from openai import OpenAI
-
-APP_TITLE = "ZEN Executive Briefing Desk"
-DEFAULT_MODEL = "gpt-4.1-mini"  # learners can change this to gpt-5.x when available
-
-SYSTEM_PROMPT = """
-You are the ZEN Executive Briefing Copilot.
-
-You help busy professionals transform raw inputs (notes, transcripts, emails, web research,
-brain dumps) into crisp, actionable, and well-structured outputs.
-
-You ALWAYS:
-- Use clear, direct language.
-- Avoid jargon unless explicitly requested.
-- Highlight decisions, risks, and next steps.
-- Write in a tone that matches the selected tone parameter.
-- Match the requested format (summary, email, LinkedIn post, action plan, etc.).
-
-If the user input is chaotic, infer structure and clean it up.
-If the user provides bullet points, upgrade them into a polished professional artifact.
-"""
-
-BRIEF_TYPES = {
-    "Executive Summary": "Write a concise executive summary with key points and a short conclusion.",
-    "Action Plan": "Produce a numbered action plan with owners (generic if not specified) and timelines.",
-    "Decision Brief": "Write a decision brief: context, options, trade-offs, recommendation.",
-    "Polished Email": "Write a professional email body only, no greeting name unless specified.",
-    "LinkedIn Post": "Write a compelling LinkedIn post, optimized for engagement but not cringe.",
-    "Meeting Minutes": "Produce structured meeting minutes: attendees, topics, decisions, action items.",
+import openai
+
+# --------------- PRESETS ---------------
+
+PRESETS = {
+    "Executive Summary":
+        {
+            "system": (
+                "You are an executive briefing assistant. "
+                "Turn complex text into a concise 3–5 bullet executive summary "
+                "with clear insights and recommendations."
+            ),
+            "user": "Paste a long report, transcript, or document here and I'll summarize it."
+        },
+    "Polished Email Reply":
+        {
+            "system": (
+                "You are a professional communications assistant. "
+                "Write clear, concise, and friendly emails suitable for executives. "
+                "Preserve tone and key details."
+            ),
+            "user": "Paste the email you received and a few notes on how you want to respond."
+        },
+    "Meeting Notes → Action Plan":
+        {
+            "system": (
+                "You convert messy meeting notes into a clear action-oriented summary. "
+                "Always include: 1) Key decisions, 2) Action items with owners and timelines, "
+                "3) Risks or next steps."
+            ),
+            "user": "Paste your rough meeting notes or bullet points and I'll structure them."
+        },
+    "Idea Generator / Brainstorm":
+        {
+            "system": (
+                "You are a creative strategist. Generate structured ideas with titles, "
+                "descriptions, and next steps."
+            ),
+            "user": "Describe what you want to build, improve, or launch. I'll propose ideas."
+        },
 }
 
-TONES = [
-    "Neutral / Standard",
-    "Confident",
-    "Concise",
-    "Visionary",
-    "Friendly",
-]
-
-LENGTH_OPTIONS = {
-    "Very short": "Target 150–250 words.",
-    "Short": "Target 250–400 words.",
-    "Medium": "Target 400–650 words.",
-    "Long": "Target 650–900 words.",
-}
+# --------------- CORE CHAT LOGIC ---------------
 
+def run_completion(api_key, model, system_prompt, user_prompt, temperature, history):
+    if not api_key:
+        raise gr.Error("🔑 Please save your API key first.")
 
-def get_client(effective_key: str) -> OpenAI:
-    return OpenAI(api_key=effective_key)
+    if not user_prompt.strip():
+        raise gr.Error("✏️ Add a prompt or select a preset before running.")
 
+    openai.api_key = api_key
+    messages = []
+    if system_prompt.strip():
+        messages.append({"role": "system", "content": system_prompt.strip()})
+    messages.append({"role": "user", "content": user_prompt.strip()})
 
-def save_api_key(user_key: str) -> Tuple[str, str]:
-    """
-    Save the API key into session state.
-    NOTE: In Spaces, this is only in-memory per session, not persisted to disk.
-    """
-    user_key = (user_key or "").strip()
-    if not user_key:
-        return "⚠️ No API key provided. Please paste a valid key.", ""
-    # Very light validation: pattern check, but don't over-assume.
-    if not (user_key.startswith("sk-") and len(user_key) > 20):
-        # Still accept it, just warn.
-        return "⚠️ Key saved, but it doesn't look like a standard OpenAI key. Double-check if requests fail.", user_key
-    return "✅ API key saved for this session.", user_key
+    # GPT-5 models use max_completion_tokens instead of max_tokens
+    response = openai.chat.completions.create(
+        model=model,
+        messages=messages,
+        temperature=float(temperature),
+        max_completion_tokens=1200,
+    )
 
+    answer = response.choices[0].message.content.strip()
+    history = history or []
+    history.append(("You", user_prompt))
+    history.append(("Assistant", answer))
+    return history, answer
 
-def build_user_prompt(
-    brief_type: str,
-    tone: str,
-    length_choice: str,
-    title: str,
-    context: str,
-    extra_instructions: str,
-) -> str:
-    type_instruction = BRIEF_TYPES.get(brief_type, "")
-    length_instruction = LENGTH_OPTIONS.get(length_choice, "")
-    tone_instruction = f"Desired tone: {tone}."
 
-    title_part = f"Working title or subject: {title.strip()}\n" if title.strip() else ""
+# --------------- HELPERS ---------------
 
-    extra_part = f"\nAdditional instructions from user:\n{extra_instructions.strip()}\n" if extra_instructions.strip() else ""
+def save_api_key(raw_key):
+    cleaned = raw_key.strip()
+    if not cleaned:
+        raise gr.Error("🔐 Please paste a valid API key.")
+    if not cleaned.startswith("sk-"):
+        gr.Info("Key saved, but it doesn’t start with 'sk-'. Double-check your provider format.")
+    return cleaned, gr.update(value="", placeholder="API key saved ✔")
 
-    return f"""
-FORMAT REQUEST:
-- Brief type: {brief_type}
-- {type_instruction}
-- {tone_instruction}
-- {length_instruction}
 
-{title_part}
-Raw notes / source material:
-\"\"\"{context.strip()}\"\"\"
+def load_preset(preset_name):
+    if not preset_name or preset_name not in PRESETS:
+        return gr.update(), gr.update()
+    preset = PRESETS[preset_name]
+    return preset["system"], preset["user"]
 
-{extra_part}
 
-Produce only the final formatted output. Do NOT include meta commentary about what you are doing.
-"""
+# --------------- UI ---------------
 
+CUSTOM_CSS = """
+#zen-root {
+    background: radial-gradient(circle at top left, #111827, #020617 65%);
+    color: #f3f4f6;
+    font-family: 'Inter', system-ui, sans-serif;
+    min-height: 100vh;
+}
+.zen-card {
+    border-radius: 20px;
+    background: rgba(17, 24, 39, 0.6);
+    border: 1px solid rgba(255,255,255,0.08);
+    backdrop-filter: blur(16px);
+    box-shadow: 0 0 40px rgba(0,0,0,0.35);
+    padding: 1.75rem;
+}
+.zen-pill {
+    border-radius: 9999px;
+    padding: 0.3rem 0.9rem;
+    border: 1px solid rgba(255,255,255,0.15);
+    color: #a5b4fc;
+    font-size: 0.75rem;
+    text-transform: uppercase;
+    letter-spacing: 0.08em;
+}
+.zen-title {
+    font-size: 1.9rem;
+    font-weight: 700;
+    letter-spacing: -0.02em;
+}
+"""
 
-def generate_brief(
-    stored_key: str,
-    inline_key: str,
-    model: str,
-    brief_type: str,
-    tone: str,
-    length_choice: str,
-    title: str,
-    context: str,
-    extra_instructions: str,
-):
-    # Resolve effective key priority: stored key -> inline key -> env variable
-    effective_key = (stored_key or "").strip() or (inline_key or "").strip() or os.getenv("OPENAI_API_KEY", "").strip()
-
-    if not effective_key:
-        return "❌ No API key available. Please paste your key and click 'Save API Key' first."
-
-    if not context.strip():
-        return "⚠️ Please paste some notes, text, or content to transform."
-
-    user_prompt = build_user_prompt(
-        brief_type=brief_type,
-        tone=tone,
-        length_choice=length_choice,
-        title=title,
-        context=context,
-        extra_instructions=extra_instructions,
+with gr.Blocks(css=CUSTOM_CSS, elem_id="zen-root", fill_height=True, title="ZEN Promptboard — GPT-5") as demo:
+    gr.Markdown(
+        """
+<div class="zen-card">
+  <div class="zen-pill">ZEN VANGUARD • GPT-5 CONSOLE</div>
+  <div class="zen-title" style="margin-top: 0.8rem;">Professional AI Promptboard</div>
+  <div style="margin-top: 0.5rem; color:#9ca3af;">
+    Transform unstructured content into polished professional outputs — with GPT-5 as your assistant.
+  </div>
+</div>
+        """
     )
 
-    try:
-        client = get_client(effective_key)
-        response = client.chat.completions.create(
-            model=model or DEFAULT_MODEL,
-            messages=[
-                {"role": "system", "content": SYSTEM_PROMPT},
-                {"role": "user", "content": user_prompt},
-            ],
-            temperature=0.5,
-            max_tokens=1200,
-        )
-        content = response.choices[0].message.content
-        if not content or not content.strip():
-            return "⚠️ Model returned an empty response. Try again or adjust your prompt."
-        return content.strip()
-    except Exception as e:
-        return f"❌ Error while calling the model:\n\n`{e}`"
-
-
-def build_ui():
-    with gr.Blocks(fill_height=True, theme=gr.themes.Soft(), title=APP_TITLE) as demo:
-        gr.Markdown(
-            f"# 🧠 {APP_TITLE}\n"
-            "Turn messy notes, transcripts, and drafts into clean executive artifacts.\n"
-            "Paste your content, pick a format, and let your model do the heavy lifting."
-        )
-
-        api_key_state = gr.State(value="")
+    api_key_state = gr.State("")
 
-        with gr.Row():
-            with gr.Column(scale=1):
-                gr.Markdown("### 🔐 API & Model")
-
-                inline_key = gr.Textbox(
-                    label="OpenAI / GPT API Key (optional if environment variable is set)",
-                    placeholder="sk-...",
+    with gr.Row(equal_height=True):
+        with gr.Column(scale=1):
+            with gr.Group():
+                gr.Markdown("### 🔐 Connect")
+                api_key_input = gr.Textbox(
+                    label="API Key",
+                    placeholder="Paste your GPT-5 key here (not stored on server)",
                     type="password",
-                    lines=1,
-                )
-                save_button = gr.Button("💾 Save API Key for This Session", variant="secondary")
-                save_status = gr.Markdown("")
-
-                model = gr.Textbox(
-                    label="Model name",
-                    value=DEFAULT_MODEL,
-                    placeholder="e.g., gpt-4.1-mini or gpt-5.x when available",
-                )
-
-                brief_type = gr.Radio(
-                    label="Brief type",
-                    choices=list(BRIEF_TYPES.keys()),
-                    value="Executive Summary",
-                )
-
-                tone = gr.Radio(
-                    label="Tone",
-                    choices=TONES,
-                    value="Neutral / Standard",
-                )
-
-                length_choice = gr.Radio(
-                    label="Length",
-                    choices=list(LENGTH_OPTIONS.keys()),
-                    value="Short",
                 )
-
-            with gr.Column(scale=2):
-                gr.Markdown("### 📝 Input")
-
-                title = gr.Textbox(
-                    label="Title / Subject (optional)",
-                    placeholder="e.g., Q4 AI Initiative Brief, Client Proposal Follow-up, etc.",
+                with gr.Row():
+                    save_btn = gr.Button("Save Key", variant="primary")
+                    key_status = gr.Markdown("Key not saved yet.")
+
+                model_name = gr.Textbox(
+                    label="Model ID",
+                    value="gpt-5",
+                    info="Default: GPT-5. You may switch to gpt-4o or gpt-4.1-mini if needed.",
                 )
-
-                context = gr.Textbox(
-                    label="Paste your notes, transcript, draft, or source content",
-                    placeholder="Drop in meeting notes, raw transcript, bullet points, or an ugly draft email...",
-                    lines=14,
+                temperature = gr.Slider(
+                    0.0, 1.2, 0.5, 0.05, label="Creativity"
                 )
 
-                extra_instructions = gr.Textbox(
-                    label="Extra instructions (optional)",
-                    placeholder="Anything special? e.g., 'Emphasize cost savings', 'Avoid technical terms', 'Target directors-level audience'.",
-                    lines=3,
+            with gr.Group():
+                gr.Markdown("### ⚡ Presets")
+                preset_radio = gr.Radio(
+                    label="Quick workflows",
+                    choices=list(PRESETS.keys()),
+                    interactive=True,
                 )
 
-        generate_button = gr.Button("⚡ Generate Brief", variant="primary")
-
-        gr.Markdown("### 📄 Output")
-        output = gr.Markdown("Your formatted brief will appear here.")
+        with gr.Column(scale=2):
+            system_box = gr.Textbox(
+                label="System / Role Instructions",
+                placeholder="How should the AI behave?",
+                lines=4,
+            )
+            user_box = gr.Textbox(
+                label="Your Prompt or Notes",
+                placeholder="Paste text, meeting notes, or raw input...",
+                lines=8,
+            )
+            run_btn = gr.Button("Run Prompt", variant="primary")
+            output_md = gr.Markdown("Output will appear here.")
+            history_chat = gr.Chatbot(label="History", height=300)
+
+    # --- Wiring ---
+
+    def _save_and_ack(key):
+        stored_key, placeholder_update = save_api_key(key)
+        return stored_key, "✅ API key stored for this session.", placeholder_update
+
+    save_btn.click(
+        _save_and_ack,
+        inputs=[api_key_input],
+        outputs=[api_key_state, key_status, api_key_input],
+    )
 
-        # Wiring
-        save_button.click(
-            fn=save_api_key,
-            inputs=[inline_key],
-            outputs=[save_status, api_key_state],
-        )
+    preset_radio.change(
+        fn=load_preset,
+        inputs=[preset_radio],
+        outputs=[system_box, user_box],
+    )
 
-        generate_button.click(
-            fn=generate_brief,
-            inputs=[
-                api_key_state,
-                inline_key,
-                model,
-                brief_type,
-                tone,
-                length_choice,
-                title,
-                context,
-                extra_instructions,
-            ],
-            outputs=[output],
+    def _run(api_state, model, system, user, temp, history):
+        history, answer = run_completion(
+            api_key=api_state,
+            model=model,
+            system_prompt=system,
+            user_prompt=user,
+            temperature=temp,
+            history=history,
         )
+        return history, answer
 
-    return demo
-
+    run_btn.click(
+        _run,
+        inputs=[api_key_state, model_name, system_box, user_box, temperature, history_chat],
+        outputs=[history_chat, output_md],
+    )
 
 if __name__ == "__main__":
-    demo = build_ui()
     demo.launch()