Update app.py

app.py

@@ -1,23 +1,15 @@
 import gradio as gr
-import random
 import time
 import openai
 
 from llama_index import StorageContext, load_index_from_storage
-import os
 
 import pandas as pd
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
-passwd = os.getenv("PASSWD_SECRET")
-
 df = pd.read_csv("original_huberman.csv")
 
 storage_context = StorageContext.from_defaults(persist_dir="./storage")
 # load index
-index = load_index_from_storage(storage_context)
-
-query_engine = index.as_query_engine(similarity_top_k=4)
 
 def get_podcast_and_youtube(response):
     podcasts = []
@@ -32,6 +24,15 @@ with gr.Blocks() as demo:
     gr.Markdown("<h1><center>HuberChat</center></h1>")
     gr.Markdown("<p align='center'><img src='https://yt3.googleusercontent.com/5ONImZvpa9_hYK12Xek2E2JLzRc732DWsZMX2F-AZ1cTutTQLBuAmcEtFwrCgypqJncl5HrV2w=s900-c-k-c0x00ffffff-no-rj' height='50' width='95'></p>")
     gr.Markdown("<p align='center' style='font-size: 20px;'>Hi! I am Andrew HuberChat, a chatbot trained to answer neurobiology.</p>")
+    gr.Markdown("<p align='center' style='font-size: 20px;'>Disclaimer: this is a fan-made project to highlight the work of Andrew Huberman. To support this project, please have a look at <a href='https://hubermanlab.com/'>Huberman Lab</a>.</p>")
+    with gr.Row().style():
+        with gr.Column(scale=1.0):
+            openai_api_key = gr.Textbox(
+                show_label=False,
+                placeholder="Set your OpenAI API key here.",
+                lines=1,
+                type="password"
+            ).style(container=False)
     with gr.Row().style():
         with gr.Column(scale=0.85):
             msg = gr.Textbox(
@@ -44,7 +45,7 @@ with gr.Blocks() as demo:
     gr.Examples(
         examples=["What is love?",
                   "Why should I get sunlight exposure?",
-                  "How to be more motivated?"
+                  "What are the benefits of walks after lunch?"
                 ],
         inputs=msg
     )
@@ -52,7 +53,11 @@ with gr.Blocks() as demo:
     
     clear = gr.Button("Clear")
 
-    def respond(message, chat_history):
+    def respond(openai_api_key, message, chat_history):
+        openai.api_key = openai_api_key
+        index = load_index_from_storage(storage_context)
+        query_engine = index.as_query_engine(similarity_top_k=3)
+
         response = query_engine.query(message)
         bot_message = response.response
         for i, row in get_podcast_and_youtube(response).iterrows():
@@ -61,9 +66,9 @@ with gr.Blocks() as demo:
         time.sleep(1)
         return "", chat_history
 
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    btn2.click(respond, [msg, chatbot], [msg, chatbot])
+    msg.submit(respond, [openai_api_key, msg, chatbot], [msg, chatbot])
+    btn2.click(respond, [openai_api_key, msg, chatbot], [msg, chatbot])
     clear.click(lambda: None, None, chatbot, queue=False)
 
 if __name__ == "__main__":
-    demo.launch(auth=("huberchat", passwd))
+    demo.launch()