|
|
|
import os |
|
import keyfile |
|
import warnings |
|
warnings.filterwarnings("ignore") |
|
|
|
|
|
from langchain_google_genai import ChatGoogleGenerativeAI |
|
from langchain.schema import HumanMessage , SystemMessage , ATMessage |
|
|
|
|
|
st.set_page_config(page_title = "Magical Healer") |
|
st.header("welcome, what help do you need?") |
|
|
|
|
|
|
|
|
|
if "sessionMessages" not in st.session_state: |
|
st.session_state.sessionMessage = [ |
|
SystemMessage(content = "You are a medieva magical healer known for your peculiar sarcasm.") |
|
] |
|
|
|
|
|
os.environ["GOOGLE_API_KEY"] = keyfile.GOOGLEKEY |
|
|
|
|
|
|
|
def load_answer(question): |
|
st.session_state.sessionMessage.append(HumanMessage(content = question)) |
|
assistant_answer = llm.invoke(st.session_state.sessionMessages) |
|
st.session_state.sessionMessages.append(AIMessge(content = assistant_answer)) |
|
return assistant_answer.content |
|
|
|
|
|
|
|
def get_text(): |
|
input_text = st.text_input("You: ", key = input) |
|
return input_text |
|
|
|
|
|
|