Update app.py

app.py

@@ -228,7 +228,7 @@ def chatbot(policy_name_dd,contract_text,progress=gr.Progress()):
         out_file.write(trimmed_input)
     """
     gpt_model = "gpt-4-1106-preview"
-    response = openai.ChatCompletion.create(
+    response = openai.chat.completions.create(
         model=gpt_model,
         temperature=0,
         messages=[
@@ -243,18 +243,18 @@ def chatbot(policy_name_dd,contract_text,progress=gr.Progress()):
         ]
     )
 
-    gpt_response = response.choices[0].message["content"]
+    gpt_response = response.choices[0].message.content
 
     tokens_used = response.usage
-    if gpt_model=="gpt-4":
-        question_cost = (tokens_used.get('total_tokens', 0) / 1000) * .03
-        prompt_tokens = tokens_used.get('prompt_tokens',)
-        completion_tokens = tokens_used.get('completion_tokens', 0)
-  
+    if gpt_model == "gpt-4":
+        question_cost = (tokens_used.total_tokens / 1000) * .03
+        prompt_tokens = tokens_used.prompt_tokens
+        completion_tokens = tokens_used.completion_tokens
+
     else:
-        prompt_tokens = tokens_used.get('prompt_tokens',)
-        completion_tokens = tokens_used.get('completion_tokens', 0)        
-        question_cost = ((prompt_tokens / 1000 ) * .01) + ((completion_tokens / 1000 ) * .03)
+        prompt_tokens = tokens_used.prompt_tokens
+        completion_tokens = tokens_used.completion_tokens
+        question_cost = ((prompt_tokens / 1000) * .01) + ((completion_tokens / 1000) * .03)
 
     """
     with open('response.txt', 'w', encoding='utf-8') as out_file: