Update app.py

app.py

@@ -1,6 +1,5 @@
 import json
 import textwrap
-import io
 from typing import Dict, Any, List, Tuple
 
 import gradio as gr
@@ -20,17 +19,23 @@ def call_chat_completion(
     system_prompt: str,
     user_prompt: str,
     temperature: float = 0.3,
-    max_tokens: int = 1800,
+    max_completion_tokens: int = 1800,
 ) -> str:
     """
     Generic OpenAI-compatible chat completion call using HTTP.
-    Supports providers that mimic the /v1/chat/completions API.
+
+    Primary path:
+      - Uses `max_completion_tokens` (new OpenAI-style param).
+
+    Fallback:
+      - If the provider complains that `max_completion_tokens` is unsupported,
+        retry once using legacy `max_tokens` instead.
     """
     if not api_key:
         raise ValueError("API key is required.")
 
     if not base_url:
-        base_url = "https://api.openai.com"  # default
+        base_url = "https://api.openai.com"
 
     url = base_url.rstrip("/") + "/v1/chat/completions"
 
@@ -39,17 +44,28 @@ def call_chat_completion(
         "Content-Type": "application/json",
     }
 
-    payload = {
-        "model": model,
-        "temperature": temperature,
-        "max_tokens": max_tokens,
-        "messages": [
-            {"role": "system", "content": system_prompt},
-            {"role": "user", "content": user_prompt},
-        ],
-    }
+    def _payload(use_new_param: bool) -> Dict[str, Any]:
+        base = {
+            "model": model,
+            "temperature": temperature,
+            "messages": [
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": user_prompt},
+            ],
+        }
+        if use_new_param:
+            base["max_completion_tokens"] = max_completion_tokens
+        else:
+            base["max_tokens"] = max_completion_tokens
+        return base
+
+    # First attempt: new param
+    resp = requests.post(url, headers=headers, json=_payload(True), timeout=60)
+
+    # If provider doesn't support max_completion_tokens, fallback to max_tokens
+    if resp.status_code == 400 and "max_completion_tokens" in resp.text:
+        resp = requests.post(url, headers=headers, json=_payload(False), timeout=60)
 
-    resp = requests.post(url, headers=headers, json=payload, timeout=60)
     if resp.status_code != 200:
         raise RuntimeError(
             f"LLM API error: {resp.status_code} - {resp.text[:400]}"
@@ -69,14 +85,14 @@ def call_chat_completion(
 SOP_SYSTEM_PROMPT = """
 You are an expert operations consultant and technical writer.
 
-You generate **clear, professional, and implementation-ready Standard Operating Procedures (SOPs)**.
+You generate clear, professional, implementation-ready Standard Operating Procedures (SOPs).
 
 The user will give:
 - A title or short description of the process
 - Context about the organization or scenario
 - Optional industry, tone, and detail level
 
-You MUST respond **strictly as JSON**, with no extra commentary, using this schema:
+You MUST respond strictly as JSON, with no extra commentary, using this schema:
 
 {
   "title": "string",
@@ -143,17 +159,18 @@ def parse_sop_json(raw_text: str) -> Dict[str, Any]:
     """
     text = raw_text.strip()
 
-    # Strip markdown fences if present
+    # Strip markdown code fences if present
     if text.startswith("```"):
-        # Remove leading ```json or ```
-        text = text.split("```", 2)
-        if len(text) == 3:
-            text = text[1] if "{" in text[1] else text[2]
+        parts = text.split("```")
+        # Try to find part containing '{'
+        candidates = [p for p in parts if "{" in p]
+        if candidates:
+            text = candidates[0]
         else:
-            text = text[-1]
+            text = parts[-1]
     text = text.strip()
 
-    # Try to find the first '{' and last '}' to extract JSON
+    # Extract JSON between first '{' and last '}'
     first_brace = text.find("{")
     last_brace = text.rfind("}")
     if first_brace != -1 and last_brace != -1:
@@ -164,6 +181,7 @@ def parse_sop_json(raw_text: str) -> Dict[str, Any]:
 
 def sop_to_markdown(sop: Dict[str, Any]) -> str:
     """Render the SOP JSON into a readable Markdown document."""
+
     def bullet_list(items: List[str]) -> str:
         if not items:
             return "_None specified._"
@@ -260,6 +278,7 @@ def create_sop_steps_figure(sop: Dict[str, Any]) -> Figure:
             wrap=True,
         )
         ax.axis("off")
+        fig.tight_layout()
         return fig
 
     n = len(steps)
@@ -274,13 +293,12 @@ def create_sop_steps_figure(sop: Dict[str, Any]) -> Figure:
         owner = step.get("owner_role", "")
         desc = step.get("description", "")
 
-        # Clamp description length
         desc_short = textwrap.shorten(desc, width=120, placeholder="...")
 
-        # Draw a rounded rectangle as a "card"
         x0, x1 = 0.05, 0.95
         y0, y1 = y - 0.3, y + 0.3
 
+        # Card rectangle (using default styles)
         ax.add_patch(
             plt.Rectangle(
                 (x0, y0),
@@ -291,7 +309,8 @@ def create_sop_steps_figure(sop: Dict[str, Any]) -> Figure:
                 linestyle="-",
             )
         )
-        # Number bubble
+
+        # Step number
         ax.text(
             x0 + 0.02,
             y,
@@ -302,7 +321,7 @@ def create_sop_steps_figure(sop: Dict[str, Any]) -> Figure:
             fontweight="bold",
         )
 
-        # Title & owner
+        # Title
         ax.text(
             x0 + 0.12,
             y + 0.15,
@@ -312,6 +331,8 @@ def create_sop_steps_figure(sop: Dict[str, Any]) -> Figure:
             fontsize=11,
             fontweight="bold",
         )
+
+        # Owner
         if owner:
             ax.text(
                 x0 + 0.12,
@@ -427,7 +448,7 @@ def generate_sop_ui(
             system_prompt=SOP_SYSTEM_PROMPT,
             user_prompt=user_prompt,
             temperature=0.25,
-            max_tokens=1800,
+            max_completion_tokens=1800,
         )
 
         sop_json = parse_sop_json(raw_response)
@@ -454,13 +475,13 @@ with gr.Blocks(title="ZEN Simple SOP Builder") as demo:
 # 🧭 ZEN Simple SOP Builder
 
 Generate clean, professional Standard Operating Procedures (SOPs) from a short description.  
-Perfect for mid-career professionals who need **clarity, structure, and ownership** — fast.
+Perfect for mid-career professionals who need clarity, structure, and ownership — fast.
 
 1. Configure your API settings  
 2. Describe the process you want to document  
 3. Generate a full SOP + visual flow of the steps
 
-> Your API key is stored only in your session state and **never logged or saved to disk**.
+> Your API key is stored only in your session state and is not logged to disk.
 """
     )
 
@@ -472,7 +493,7 @@ Perfect for mid-career professionals who need **clarity, structure, and ownershi
 
             api_key_input = gr.Textbox(
                 label="LLM API Key",
-                placeholder="Enter your API key (OpenAI, compatible provider, etc.)",
+                placeholder="Enter your API key (OpenAI or compatible provider)",
                 type="password",
             )
             base_url = gr.Textbox(
@@ -488,8 +509,8 @@ Perfect for mid-career professionals who need **clarity, structure, and ownershi
 
             gr.Markdown(
                 """
-You can use **any OpenAI-compatible provider**  
-(just update the Base URL + Model Name accordingly).
+You can use any **OpenAI-compatible** provider  
+by adjusting the Base URL and Model Name.
 """
             )