Update app.py

app.py

@@ -228,7 +228,8 @@ def chatbot(policy_name_dd,contract_text,progress=gr.Progress()):
         out_file.write(trimmed_input)
     """
     gpt_model = "gpt-4-1106-preview"
-    response = openai.chat.completions.create(
+    response = openai.ChatCompletion.create(
+    #response = openai.chat.completions.create(
         model=gpt_model,
         temperature=0,
         messages=[
@@ -243,17 +244,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"]
+    #gpt_response = response.choices[0].message.content
 
     tokens_used = response.usage
     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
+        question_cost = (tokens_used.get('total_tokens',0) / 1000) * .03
+        prompt_tokens = tokens_used.get('prompt_tokens',0)
+        completion_tokens = tokens_used.get('completion_tokens',0)
 
     else:
-        prompt_tokens = tokens_used.prompt_tokens
-        completion_tokens = tokens_used.completion_tokens
+        prompt_tokens = tokens_used.get('prompt_tokens',0)
+        completion_tokens = tokens_used.get('completion_tokens',0)
         question_cost = ((prompt_tokens / 1000) * .01) + ((completion_tokens / 1000) * .03)
 
     """