Update app.py

app.py

@@ -10,22 +10,51 @@ from docx import Document
 
 def call_model(client, model_name, system_prompt, user_prompt):
 
-    res = client.chat.completions.create(
-        model=model_name,
-        temperature=0.2,
-        messages=[
-            {
-                "role": "system",
-                "content": system_prompt
-            },
-            {
-                "role": "user",
-                "content": user_prompt
-            }
-        ]
-    )
+    try:
+
+        res = client.chat.completions.create(
+            model=model_name,
+            temperature=0.2,
+            messages=[
+                {
+                    "role": "system",
+                    "content": system_prompt
+                },
+                {
+                    "role": "user",
+                    "content": user_prompt
+                }
+            ]
+        )
+
+        # Defensive checks
+        if res is None:
+
+            return "ERROR: Response is None"
+
+        if not hasattr(res, "choices"):
+
+            return "ERROR: Response has no choices"
+
+        if not res.choices:
+
+            return "ERROR: choices empty"
+
+        if res.choices[0].message is None:
 
-    return res.choices[0].message.content
+            return "ERROR: message empty"
+
+        content = res.choices[0].message.content
+
+        if content is None:
+
+            return "ERROR: content empty"
+
+        return content
+
+    except Exception as e:
+
+        return f"ERROR: {str(e)}"
 
 
 def extract_json(text):
@@ -221,11 +250,27 @@ def generate_document(api_key, raw_input):
         raw_input
     )
 
+    if "ERROR:" in thinker_json:
+
+        return (
+            thinker_json,
+            "Designer skipped",
+            None
+        )
+
     designer_json = run_designer(
         api_key,
         thinker_json
     )
 
+    if "ERROR:" in designer_json:
+
+        return (
+            thinker_json,
+            designer_json,
+            None
+        )
+
     doc_path = build_word(
         designer_json
     )