Update app.py

app.py

@@ -227,11 +227,9 @@ def chatbot(policy_name_dd,contract_text,progress=gr.Progress()):
     with open('analysis_input.txt', 'w', encoding='utf-8') as out_file:
         out_file.write(trimmed_input)
     """
-
+    gpt_model = "gpt-4-1106-preview"
     response = openai.ChatCompletion.create(
-        #model="gpt-4",
-        model="gpt-4-1106-preview",
-        # model="gpt-3.5-turbo",
+        model=gpt_model,
         temperature=0,
         messages=[
             {
@@ -248,10 +246,15 @@ def chatbot(policy_name_dd,contract_text,progress=gr.Progress()):
     gpt_response = response.choices[0].message["content"]
 
     tokens_used = response.usage
-
-    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.get('total_tokens', 0) / 1000) * .03
+        prompt_tokens = tokens_used.get('prompt_tokens',)
+        completion_tokens = tokens_used.get('completion_tokens', 0)
+  
+    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)
 
     """
     with open('response.txt', 'w', encoding='utf-8') as out_file: