Update app.py

app.py

@@ -83,9 +83,15 @@ def generate_email_content(api_key, prompt):
             max_tokens=1024,
             stream=False  # Disable streaming for simplicity
         )
-        email_content = response['choices'][0]['message']['content']
-        logging.info("Email content generated successfully.")
-        return email_content
+        
+        # Access the response content correctly based on the Nvidia API structure
+        if hasattr(response, 'choices') and len(response.choices) > 0:
+            email_content = response.choices[0].message.content
+            logging.info("Email content generated successfully.")
+            return email_content
+        else:
+            logging.error("Error: No choices found in the response.")
+            return "Error generating email content: No valid choices."
     except Exception as e:
         logging.error(f"Error generating email content: {e}")
         return "Error generating email content."