Update app.py

app.py

@@ -1,66 +1,50 @@
-import langchain
-#from pydantic import BaseModel, model_validator
-from langchain.embeddings.openai import OpenAIEmbeddings
-# from langchain.vectorstores import Chroma
-from langchain.vectorstores import FAISS
-from langchain.text_splitter import CharacterTextSplitter
-from langchain.llms import OpenAI
-from langchain.chains import VectorDBQA
-from langchain.chains import RetrievalQA
-from langchain.document_loaders import DirectoryLoader
-from langchain.chains import ConversationalRetrievalChain
-from langchain.memory import ConversationBufferMemory
-from langchain.evaluation.qa import QAGenerateChain
-import magic
-import os
+from typing import List, Optional
+
+from pydantic import BaseModel, BaseSettings, SecretStr
 import streamlit as st
-from streamlit_chat import message
 
-st.title("Welcome to ExamPrepBot")
+# Slide 3: Basic Model
+
+class User(BaseModel):
+    id: int
+    name: str = "Jane Doe"
+
+data = {"id": 19, "name": "Fanilo", "age": 179}
+user = User(**data)
+st.write(user)
+
+data = {"id": "Fanilo", "name": 42}
+user = User(**data)
+#st.write(user)
+
+# Slide 4: Hierarchical Model
 
-if 'responses' not in st.session_state:
-    st.session_state['responses'] = ["How can I assist you?"]
+class Address(BaseModel):
+    city: str
+    street: Optional[str]
 
-if 'requests' not in st.session_state:
-    st.session_state['requests'] = []
 
-openai_api_key = os.getenv("OPENAI_API_KEY", "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")    
-embeddings = OpenAIEmbeddings(openai_api_key=openai_api_key)
-new_db = FAISS.load_local("faiss_govt_RCV", embeddings)
-llm = OpenAI(openai_api_key=openai_api_key, temperature=0.0)
+class User(BaseModel):
+    id: int
+    name: str
+    addresses: List[Address]
 
-# if 'buffer_memory' not in st.session_state:
-memory= ConversationBufferMemory(memory_key="chat_history", return_messages=True)
-retriever = new_db.as_retriever()
-chain = ConversationalRetrievalChain.from_llm(llm=llm, chain_type="stuff", memory= memory,retriever=retriever, verbose=False)
 
-# container for chat history
-response_container = st.container()
-# container for text box
-textcontainer = st.container()
+data = {
+    "id": 42,
+    "name": "Fanilo",
+    "addresses": [{"city": "Paris"}, {"city": "Tokyo", "street": "こんにちは"}],
+}
+user = User(**data)
+st.success(user.addresses[1].street)
 
+# Slide 7: Secrets
 
-with textcontainer:
-    query = st.text_input(label="Please Enter Your Prompt Here: ", placeholder="Ask me")
-    if query:
-        with st.spinner("Cooking..."):
-            # conversation_string = get_conversation_string()
-            # st.code(conversation_string)
-            # refined_query = query_refiner(conversation_string, query)
-            # st.subheader("Refined Query:")
-            # st.write(refined_query)
-            # context = find_match(refined_query)
-            # print(context)  
-            response = chain.run(query)
-        st.session_state.requests.append(query)
-        st.session_state.responses.append(response) 
-with response_container:
-    if st.session_state['responses']:
+class Settings(BaseSettings):
+    auth_key: SecretStr
+    api_key: str
 
-        for i in range(len(st.session_state['responses'])):
-            message(st.session_state['responses'][i],key=str(i))
-            if i < len(st.session_state['requests']):
-                message(st.session_state["requests"][i], is_user=True,key=str(i)+ '_user')
+    class Config:
+        env_file = "settings.env"
 
-# with st.expander('Message history'):
-#     st.info(memory.buffer)
+st.write(Settings().dict())