Update app.py

app.py

@@ -5,24 +5,23 @@ import streamlit as st
 def generate_email(api_key, email_type, details):
     openai.api_key = api_key  # Set OpenAI API key provided by the user
     
-    prompt = f"""
-    I need to write a professional email for the following situation:
-    Email Type: {email_type}
-    Details: {details}
-    Please generate the body of the email, including suggestions for the subject line and sign-off. Ensure the email tone is professional and polite.
-    """
+    # Define the conversation in the chat-based format
+    messages = [
+        {"role": "system", "content": "You are a helpful assistant that writes professional emails."},
+        {"role": "user", "content": f"I need to write a professional email for the following situation:\nEmail Type: {email_type}\nDetails: {details}\nPlease generate the body of the email, including suggestions for the subject line and sign-off. Ensure the email tone is professional and polite."}
+    ]
     
     try:
-        # Request to OpenAI API
-        response = openai.Completion.create(
-            engine="gpt-4o",  # Using OpenAI's GPT-3 model (adjust for specific LLM as needed)
-            prompt=prompt,
+        # Request to OpenAI API (using chat/completions endpoint)
+        response = openai.ChatCompletion.create(
+            model="gpt-4o",  # You can replace this with any other chat-based model like `gpt-4`
+            messages=messages,
             max_tokens=1200,
             temperature=0.7
         )
         
         # Return the generated email content
-        return response.choices[0].text.strip()
+        return response.choices[0].message['content'].strip()
     except Exception as e:
         return f"Error: {str(e)}"