Update app.py

app.py

@@ -16,15 +16,15 @@ def generate_legal_document(doc_type, terms, party_1, party_2):
     Please provide a professional legal document based on these details.
     """
     
-    # Call OpenAI's GPT-3 model to generate the document
-    response = openai.Completion.create(
-        model="text-davinci-003",  # Use a model suited for legal text
-        prompt=prompt,
+    # Call OpenAI's new chat completion API
+    response = openai.chat_completions.create(
+        model="gpt-4",  # or "gpt-3.5-turbo"
+        messages=[{"role": "user", "content": prompt}],
         max_tokens=1500
     )
     
     # Return the generated text
-    return response.choices[0].text.strip()
+    return response['choices'][0]['message']['content']
 
 # Streamlit UI to collect user inputs
 st.title("Legal Document Generator")