Update app.py

app.py

@@ -14,32 +14,43 @@ client = OpenAI(api_key = key)
 def clear():
     return [None, [], None]
 
-def chat(prompt, user_window, pwd_window, past, response):
+def updatePassword(txt):
+    return [txt, None]
+
+def chat(prompt, user_window, pwd_window, past, response, gptModel):
     if user_window==uname and pwd_window==pwd:
         past.append({"role":"user", "content":prompt})
-        completion = client.chat.completions.create(model="gpt-4o-mini",
+        completion = client.chat.completions.create(model=gptModel,
                                             messages=past)
         reply = completion.choices[0].message.content
         response += "\n\nYOU: " + prompt + "\nGPT: " + reply
+        if len(response) > 40000:
+            response += "\n\nTHIS DIALOG IS GETTING TOO LONG. PLEASE CLEAR IT."
         past.append({"role":"assistant", "content": reply})
-        return [past, response]
+        return [past, response, None, None]
     else:
-        return [[], "User name and/or password are incorrect"]
+        return [[], "User name and/or password are incorrect", prompt, pwd_window]
 
 with gr.Blocks() as demo:
     history = gr.State([])
-    dialog = gr.State()
-    heading = gr.Label(value="GPT Chat", scale=2, color="Crimson" )
+    password = gr.State("")
+    model = gr.State("gpt-3.5-turbo")
+    gr.Markdown('# GPT Chat')
+    gr.Markdown('Enter user name & password then enter prompt and click submit button.  GPT 3.5 is cheaper but GPT 4o may perform better.')
+    # heading = gr.Label(value="GPT Chat", scale=2, color="Crimson" )
     with gr.Row():
         user_window = gr.Textbox(label = "User Name")
         pwd_window = gr.Textbox(label = "Password")
+        pwd_window.blur(updatePassword, pwd_window, [password, pwd_window])
     with gr.Row():
-        clear_button = gr.Button(value="Clear")
-        submit_window = gr.Button(value="Submit")
+        clear_button = gr.Button(value="Clear Dialog")
+        gpt_chooser=gr.Radio(choices=[("GPT-3.5","gpt-3.5-turbo"),("GPT-4o","gpt-4o-mini")],
+                             value="gpt-3.5-turbo", label="GPT Model", interactive=True)
+        submit_window = gr.Button(value="Submit Prompt/Question")
     prompt_window = gr.Textbox(label = "Prompt or Question")
     output_window = gr.Textbox(label = "Dialog")
-    submit_window.click(chat, inputs=[prompt_window, user_window, pwd_window, history, output_window],
-                         outputs=[history, output_window])
+    submit_window.click(chat, inputs=[prompt_window, user_window, password, history, output_window, model],
+                         outputs=[history, output_window, prompt_window, pwd_window])
     clear_button.click(clear, inputs=[], outputs=[prompt_window, history, output_window])
     
 demo.launch()