Update app.py

app.py

@@ -118,16 +118,49 @@ def summarize_articles(indices, articles_for_display):
     summary = summarize_with_huggingface(selected_articles)
     return summary
 
+PASSWORD = "pass"
+
+def check_password(username, password):
+    if username == USERNAME and password == PASSWORD:
+        return True, "Welcome!"
+    else:
+        return False, "Incorrect username or password."
+        
 # Gradio interface
 with gr.Blocks() as demo:
+    
     gr.Markdown("### PubMed Article Summarizer")
-    model_input = gr.Textbox(label="Enter the model to use", value="h2oai/h2ogpt-4096-llama2-7b-chat")
-    query_input = gr.Textbox(label="Query Keywords")
-    retmax_input = gr.Slider(minimum=1, maximum=20, value=5, step=1, label="Number of articles")
-    search_button = gr.Button("Search")
-    output_table = gr.Dataframe(headers=["PMID", "Authors", "Title","Abstract" ])
-    summarize_button = gr.Button("Summarize")
-    summary_output = gr.Textbox()
+
+    with gr.Row():
+        password_input = gr.Textbox(label="Password", type="password")
+
+        login_button = gr.Button("Login")
+        login_result = gr.Textbox(label="Login Result", interactive=False)
+
+    login_button.click(check_password, inputs=[username_input, password_input], outputs=[login_result])
+
+    with gr.Row():
+        model_input = gr.Textbox(label="Enter the model to use", value="h2oai/h2ogpt-4096-llama2-7b-chat")
+        query_input = gr.Textbox(label="Query Keywords")
+        retmax_input = gr.Slider(minimum=1, maximum=20, value=5, step=1, label="Number of articles")
+        search_button = gr.Button("Search")
+        output_table = gr.Dataframe(headers=["PMID", "Authors", "Title","Abstract" ])
+        summarize_button = gr.Button("Summarize")
+        summary_output = gr.Textbox()
+
+        model_input.visible = False
+        query_input.visible = False
+        summarize_button.visible = False
+
+    def process_login(is_success, message):
+        if is_success:
+            model_input.visible = True
+            query_button.visible = True
+            summarize_button.visible = True
+        login_result.update(value=message)
+        login_result.visible = True
+    
+    login_button.click(check_password, inputs=[username_input, password_input], outputs=[process_login])
 
     def update_output_table(query, retmax):
         df = search_pubmed(query, retmax)