Update app.py

app.py

@@ -1,59 +1,162 @@
 import os
+import json
 import gradio as gr
+import requests
 
-from openai import OpenAI
-
+# ---- Secrets (Space Settings → Secrets) ----
 HF_TOKEN = os.getenv("HF_TOKEN", "")
-HF_MODEL = os.getenv("HF_MODEL", "Qwen/Qwen2.5-7B-Instruct")
-
-client = OpenAI(base_url="https://router.huggingface.co/v1", api_key=HF_TOKEN)
-
-SYSTEM = (
-    "You are a senior enterprise AE specializing in converting open-source platform usage into "
-    "enterprise engagements. No vendor impersonation. No private data claims.\n\n"
-    "Output exactly these headings:\n"
-    "1. Account Snapshot\n"
-    "2. Usage Hypotheses\n"
-    "3. Persona Mapping\n"
-    "4. Outbound Campaign Design\n"
-    "5. Message Examples\n"
-    "6. How This Scales\n\n"
-    "Under Message Examples: one technical-user message + one decision-maker message.\n"
-    "Tone: senior AE. No emojis."
-)
-
-def run(account: str) -> str:
-    account = (account or "").strip()
-    if not account:
-        return "Please input an account name or org URL."
+DEFAULT_MODEL = os.getenv("HF_MODEL", "Qwen/Qwen2.5-7B-Instruct")
+
+# HF Router (OpenAI-compatible)
+ROUTER_URL = "https://router.huggingface.co/v1/chat/completions"
+
+SYSTEM_INSTRUCTIONS = """
+You are a senior enterprise Account Executive specializing in converting open-source platform usage into enterprise-grade commercial engagements.
+
+Rules:
+- Do NOT represent any specific vendor.
+- Do NOT claim access to private/internal data.
+- Anchor everything in public signals and state assumptions clearly.
+- Avoid generic SaaS buzzwords. Be executable.
+
+You MUST output exactly these headings:
+
+1. Account Snapshot
+2. Usage Hypotheses
+3. Persona Mapping
+4. Outbound Campaign Design
+5. Message Examples
+6. How This Scales
 
+Under Message Examples:
+- One technical-user message
+- One decision-maker message
+
+Tone: senior AE speaking to peers. No emojis.
+""".strip()
+
+USER_TEMPLATE = """
+Account (name or org URL): {account}
+
+Optional context:
+Industry: {industry}
+Region: {region}
+Primary persona: {primary_persona}
+Secondary personas: {secondary_personas}
+Offer focus: {offer_focus}
+Goal: {goal}
+Notes: {notes}
+
+Task:
+Create a structured outbound plan for this account based on public open-source platform usage and OSS → Enterprise conversion patterns.
+""".strip()
+
+
+def call_router(model: str, system: str, user: str) -> str:
     if not HF_TOKEN:
         return (
             "Missing HF_TOKEN.\n\n"
-            "Fix: Space → Settings → Secrets → add HF_TOKEN with permission "
-            "“Make calls to Inference Providers”."
+            "Fix:\n"
+            "Space → Settings → Secrets → add HF_TOKEN with permission:\n"
+            "✅ Make calls to Inference Providers"
         )
 
+    headers = {
+        "Authorization": f"Bearer {HF_TOKEN}",
+        "Content-Type": "application/json",
+    }
+
+    payload = {
+        "model": model,
+        "messages": [
+            {"role": "system", "content": system},
+            {"role": "user", "content": user},
+        ],
+        "temperature": 0.4,
+        "max_tokens": 1400,
+    }
+
     try:
-        resp = client.chat.completions.create(
-            model=HF_MODEL,
-            messages=[
-                {"role": "system", "content": SYSTEM},
-                {"role": "user", "content": f"Account: {account}\n\nCreate the outbound plan now."},
-            ],
-            temperature=0.4,
-            max_tokens=900,
+        r = requests.post(ROUTER_URL, headers=headers, data=json.dumps(payload), timeout=60)
+        if r.status_code != 200:
+            return (
+                "Model call failed.\n\n"
+                f"HTTP {r.status_code}\n"
+                f"Response: {r.text}\n\n"
+                "Quick fixes:\n"
+                "1) Try a smaller model (e.g., Qwen/Qwen2.5-7B-Instruct or mistralai/Mistral-7B-Instruct-v0.3)\n"
+                "2) Ensure your HF_TOKEN includes 'Make calls to Inference Providers'\n"
+                "3) If the model is gated, pick a model you have access to"
+            )
+
+        data = r.json()
+        return data["choices"][0]["message"]["content"].strip()
+
+    except requests.exceptions.Timeout:
+        return (
+            "Model call timed out.\n\n"
+            "Fixes:\n"
+            "1) Use a smaller model\n"
+            "2) Reduce max_tokens\n"
         )
-        return resp.choices[0].message.content.strip()
     except Exception as e:
-        return f"Model call failed.\nModel: {HF_MODEL}\nError: {repr(e)}"
+        return f"Unexpected error calling router: {repr(e)}"
+
+
+def generate(account, industry, region, primary_persona, secondary_personas, offer_focus, goal, notes, model):
+    account = (account or "").strip()
+    if not account:
+        return "Please input an account name or org URL."
+
+    model = (model or DEFAULT_MODEL).strip()
+
+    user_prompt = USER_TEMPLATE.format(
+        account=account,
+        industry=industry or "Not provided",
+        region=region or "Not provided",
+        primary_persona=primary_persona or "Not provided",
+        secondary_personas=secondary_personas or "Not provided",
+        offer_focus=offer_focus or "Not provided",
+        goal=goal or "Convert free/community usage to enterprise engagement",
+        notes=notes or "Not provided",
+    )
+
+    return call_router(model=model, system=SYSTEM_INSTRUCTIONS, user=user_prompt)
+
+
+with gr.Blocks(title="Enterprise OSS → Outbound Ops") as app:
+    gr.Markdown(
+        "# Enterprise OSS → Outbound Ops\n"
+        "Input an **account name or public org URL** to generate a structured enterprise outbound plan."
+    )
+
+    account = gr.Textbox(
+        label="Account name or org URL",
+        placeholder="e.g., Emirates OR https://huggingface.co/emirates",
+        lines=1,
+    )
+
+    with gr.Row():
+        industry = gr.Textbox(label="Industry (optional)", placeholder="e.g., Airlines, Energy, Banking")
+        region = gr.Textbox(label="Region (optional)", placeholder="e.g., MENA, KSA, UAE, Global")
+
+    with gr.Row():
+        primary_persona = gr.Textbox(label="Primary persona (optional)", placeholder="e.g., Head of Data/AI, VP Eng, CISO")
+        secondary_personas = gr.Textbox(label="Secondary personas (optional)", placeholder="e.g., ML Eng, DS, Platform, Security")
+
+    offer_focus = gr.Textbox(label="Offer focus (optional)", placeholder="e.g., governance, SSO, audit, private collaboration")
+    goal = gr.Textbox(label="Goal (optional)", placeholder="e.g., Identify champions + secure intro call with platform owner")
+    notes = gr.Textbox(label="Notes / known public signals (optional)", lines=4)
+
+    model = gr.Textbox(label="Model", value=DEFAULT_MODEL)
+
+    run = gr.Button("Generate outbound plan")
+    output = gr.Markdown()
 
-demo = gr.Interface(
-    fn=run,
-    inputs=gr.Textbox(label="Account name or org URL"),
-    outputs=gr.Markdown(),
-    title="Enterprise OSS → Outbound Ops",
-    description="Paste an account name or org URL to generate a structured outbound plan."
-)
+    run.click(
+        fn=generate,
+        inputs=[account, industry, region, primary_persona, secondary_personas, offer_focus, goal, notes, model],
+        outputs=output,
+    )
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+app.launch(server_name="0.0.0.0", server_port=7860)