Update app.py

app.py

@@ -5,15 +5,13 @@ import os
 import re
 import requests
 
-api_key = os.getenv("OPENAIKEY")
-client = OpenAI(api_key=api_key)
-
-auth_token = os.getenv("DB_AUTH_TOKEN")
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+model = "gpt-3.5-turbo"
 
 def repo_get_all_employees_from_database():
     url = "https://api.airtable.com/v0/appopGmlHujYnd6Vw/Interviewers?maxRecords=100&view=Grid%20view"
     headers = {
-        "Authorization": auth_token
+        "Authorization": os.getenv("DB_AUTH_TOKEN")
     }
     response = requests.get(url, headers=headers)
     records = response.json()
@@ -96,22 +94,18 @@ def predict(message, history):
         history_openai_format.append({"role": "assistant", "content": assistant})
     history_openai_format.append({"role": "user", "content": prompt})
 
-    model = "gpt-3.5-turbo"
+    global model
+    
     if ("switch to gpt-3.5" in message.lower()):
         model = "gpt-3.5-turbo"
-        print("Switched to:")
-        print(model)
+        print("Switched to: {model}".format(model=model))
 
     if ("switch to gpt-4" in message.lower()):
         model = "gpt-4"
-        print("Switched to:")
-        print(model)
-
-    print("Actual model:")
-    print(model)    
+        print("Switched to: {model}".format(model=model)) 
   
     response = client.chat.completions.create(
-        model="gpt-4",
+        model=model,
         messages= history_openai_format,
         temperature=0,
         stream=True)
@@ -132,6 +126,9 @@ This is an AI Interview Team Assistant. You can ask him any questions about recr
 You can send any regular prompts you wish or pre-configured Chain-of-Thought prompts.\n
 To trigger pre-configured prompt you have to craft a prompt with next structure:
 - "{pre_configured_promt}"
+\n
+You can switch between gpt-3.5 and gpt-4 with "Switch to gpt-3.5" or "Switch to gpt-4" prompts.\n
+Language Model currently under the hood: {model}
 '''.format(pre_configured_promt=pre_configured_promt)
 
 examples = [pre_configured_promt]