deployable

app.py

@@ -1,13 +1,23 @@
+import os
+os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
+
 import weave
 
 import streamlit as st
 from rag.rag import SimpleRAGPipeline
 
+st.set_page_config(page_title="Chat with the Llama 3 paper!", page_icon="🦙", layout="centered", initial_sidebar_state="auto", menu_items=None)
+
+wandb_api_key = st.sidebar.text_input('WANDB_API_KEY', type='password')
+if len(wandb_api_key)>=10:
+    os.environ["WANDB_API_KEY"] = wandb_api_key
+else:
+    st.stop()
+
 WANDB_PROJECT = "paper_reader"
 
 weave.init(f"{WANDB_PROJECT}")
 
-st.set_page_config(page_title="Chat with the Llama 3 paper!", page_icon="🦙", layout="centered", initial_sidebar_state="auto", menu_items=None)
 st.title("Chat with the Llama 3 paper 💬🦙")
 
 @st.cache_resource(show_spinner=False)
@@ -22,7 +32,6 @@ if "rag_pipeline" not in st.session_state.keys():
 
 rag_pipeline = st.session_state["rag_pipeline"]
 
-# openai_api_key = st.sidebar.text_input('OpenAI API Key', type='password')
 
 def generate_response(query):
     response = rag_pipeline.predict(query)