Update app.py

app.py

@@ -102,11 +102,11 @@ def load_recommender(path, start_page=1):
 
 def generate_text(openAI_key, prompt, model="gpt-3.5-turbo"):
     openai.api_key = openAI_key
-    temperature=0.7
-    max_tokens=256
-    top_p=1
-    frequency_penalty=0
-    presence_penalty=0
+    temperature = 0.1
+    max_tokens = 256
+    top_p = 1
+    frequency_penalty = 0
+    presence_penalty = 0
 
     if model == "text-davinci-003":
         completions = openai.Completion.create(
@@ -119,19 +119,19 @@ def generate_text(openAI_key, prompt, model="gpt-3.5-turbo"):
         )
         message = completions.choices[0].text
     else:
-        message = openai.ChatCompletion.create(
+        response = openai.ChatCompletion.create(
             model=model,
             messages=[
                 {"role": "system", "content": "You are a helpful assistant."},
-                {"role": "assistant", "content": "Here is some initial assistant message."},
                 {"role": "user", "content": prompt}
             ],
-            temperature=.3,
+            temperature=temperature,
             max_tokens=max_tokens,
             top_p=top_p,
             frequency_penalty=frequency_penalty,
             presence_penalty=presence_penalty,
-        ).choices[0].message['content']
+        )
+        message = response['choices'][0]['message']['content']
     return message
 
   
@@ -255,16 +255,13 @@ with gr.Blocks(css="""#chatbot { font-size: 14px; min-height: 1200; }""") as dem
             # Model Selection
             model = gr.Radio(
                 choices=[
-                    'gpt-3.5-turbo', 
-                    'gpt-3.5-turbo-16k', 
-                    'gpt-3.5-turbo-0613', 
-                    'gpt-3.5-turbo-16k-0613', 
-                    'text-davinci-003',
-                    'gpt-4',
-                    'gpt-4-32k'
+                    'gpt-4o',
+                    'gpt-4o-mini',
+                    'o1-preview',
+                    'o1-mini'
                 ], 
                 label='Select Model', 
-                value='gpt-3.5-turbo'
+                value='gpt-4o'
             )
 
             # Chat Interface