Update app.py

app.py

@@ -6,8 +6,8 @@ from dotenv import load_dotenv
 
 load_dotenv()
 key = os.getenv('OPENAI_API_KEY')
-# uname = os.getenv('LOGNAME')
-# pwd = os.getenv('PASSWORD')
+uname = os.getenv('LOGNAME')
+pwd = os.getenv('PASSWORD')
 
 client = OpenAI(api_key = key)
 
@@ -15,23 +15,24 @@ def clear():
     return [None, None]
 
 
-def chat(prompt):
-    completion = client.chat.completions.create(model="gpt-4o-mini",
+def chat(prompt, user_window, pwd_window):
+    if user_window==uname and pwd_window==pwd:
+        completion = client.chat.completions.create(model="gpt-4o-mini",
                                             messages=[{"role":"user", "content":prompt}])
-    result = completion.choices[0].message.content
-    # else:
-    #     result = "User name and/or password are incorrect"
+        result = completion.choices[0].message.content
+    else:
+        result = "User name and/or password are incorrect"
     return result
 
 with gr.Blocks() as demo:
     heading = gr.Label(value="GPT Chat", scale=2, color="Crimson" )
-    # user_window = gr.Textbox(label = "User Name")
-    # pwd_window = gr.Textbox(label = "Password")
+    user_window = gr.Textbox(label = "User Name")
+    pwd_window = gr.Textbox(label = "Password")
     clear_button = gr.Button(value="Clear")
     prompt_window = gr.Textbox(label = "Prompt")
     submit_window = gr.Button(value="Submit")
     output_window = gr.Textbox(label = "Response")
-    submit_window.click(chat, inputs=[prompt_window], outputs=output_window)
+    submit_window.click(chat, inputs=[prompt_window, user_window, pwd_window], outputs=output_window)
     clear_button.click(clear, inputs=[], outputs=[prompt_window, output_window])
     
-demo.launch(auth=(os.getenv('LOGNAME'),os.getenv('PASSWORD')))
+demo.launch()