Update app.py

app.py

@@ -30,8 +30,13 @@ def generate_email(company, recipient_name, product, progress=gr.Progress()):
             response = requests.post(url, headers=headers, json=data, timeout=300)  # 5 minutes timeout
             if response.status_code == 200:
                 result = response.json()
-                email_content = result['response'].split('[/INST]')[-1].strip()
-                result_queue.put(("success", email_content))
+                # Check if 'response' is in the result and is a string
+                if 'response' in result and isinstance(result['response'], str):
+                    email_content = result['response'].split('[/INST]')[-1].strip()
+                    result_queue.put(("success", email_content))
+                else:
+                    # If 'response' is not as expected, return the whole result for debugging
+                    result_queue.put(("error", f"Unexpected response structure: {json.dumps(result, indent=2)}"))
             else:
                 result_queue.put(("error", f"Error: {response.status_code} - {response.text}"))
         except requests.Timeout: