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
 
   
@@ -248,23 +248,19 @@ with gr.Blocks(css="""#chatbot { font-size: 14px; min-height: 1200; }""") as dem
             # API Key and File Inputs
             with gr.Accordion("API Key and PDF"):
                 openAI_key = gr.Textbox(label='Enter your OpenAI API key here', type='password')
-                url = gr.Textbox(label='Enter PDF URL here (Example: https://arxiv.org/pdf/1706.03762.pdf)')
+                url = gr.Textbox(label='Enter PDF URL here (Example: https://arxiv.org/pdf/1706.03762.pdf ; https://link.springer.com/content/pdf/10.1007/s10614-022-10325-8.pdf)')
                 gr.Markdown("<center><h4>OR<h4></center>")
                 file = gr.File(label='Upload your PDF/Research Paper/Book here', file_types=['.pdf'])
             
             # 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-4o-mini',
+                    'gpt-4o',
                     'gpt-4',
-                    'gpt-4-32k'
                 ], 
                 label='Select Model', 
-                value='gpt-3.5-turbo'
+                value='gpt-4o-mini'
             )
 
             # Chat Interface
@@ -333,6 +329,4 @@ with gr.Blocks(css="""#chatbot { font-size: 14px; min-height: 1200; }""") as dem
         [msg, chatbot]
     )
 
-demo.launch()
-
-
+demo.launch()