Update app.py

app.py

@@ -31,12 +31,12 @@ 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()
-                # 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()
+                if 'response' in result and isinstance(result['response'], list) and len(result['response']) > 0:
+                    full_response = result['response'][0]
+                    # Extract the email content after [/INST]
+                    email_content = full_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}"))