Spaces:
Running
Running
Update Chatbot.py
Browse files- Chatbot.py +32 -32
Chatbot.py
CHANGED
@@ -1,33 +1,33 @@
|
|
1 |
-
import streamlit as st
|
2 |
-
from dotenv import load_dotenv
|
3 |
-
load_dotenv()
|
4 |
-
import os
|
5 |
-
import google.generativeai as genai
|
6 |
-
|
7 |
-
os.getenv("GOOGLE_API_KEY")
|
8 |
-
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
9 |
-
def show():
|
10 |
-
st.header("Chatbot")
|
11 |
-
model=genai.GenerativeModel('gemini-1.
|
12 |
-
chat=model.start_chat(history=[])
|
13 |
-
if 'chat_history' not in st.session_state:
|
14 |
-
st.session_state['chat_history'] = []
|
15 |
-
input=st.text_input("Input Prompt: ",key="input")
|
16 |
-
|
17 |
-
submit=st.button("Submit")
|
18 |
-
|
19 |
-
if submit and input:
|
20 |
-
|
21 |
-
|
22 |
-
response=chat.send_message(input,stream=True)
|
23 |
-
st.session_state['chat_history'].append(("You", input))
|
24 |
-
st.subheader("The Response is")
|
25 |
-
for chunk in response:
|
26 |
-
st.write(chunk.text)
|
27 |
-
st.session_state['chat_history'].append(("Bot", chunk.text))
|
28 |
-
|
29 |
-
|
30 |
-
st.subheader("The Chat History is")
|
31 |
-
|
32 |
-
for role, text in st.session_state['chat_history']:
|
33 |
st.write(f"{role}: {text}")
|
|
|
1 |
+
import streamlit as st
|
2 |
+
from dotenv import load_dotenv
|
3 |
+
load_dotenv()
|
4 |
+
import os
|
5 |
+
import google.generativeai as genai
|
6 |
+
|
7 |
+
os.getenv("GOOGLE_API_KEY")
|
8 |
+
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
9 |
+
def show():
|
10 |
+
st.header("Chatbot")
|
11 |
+
model=genai.GenerativeModel('gemini-1.5-pro-latest')
|
12 |
+
chat=model.start_chat(history=[])
|
13 |
+
if 'chat_history' not in st.session_state:
|
14 |
+
st.session_state['chat_history'] = []
|
15 |
+
input=st.text_input("Input Prompt: ",key="input")
|
16 |
+
|
17 |
+
submit=st.button("Submit")
|
18 |
+
|
19 |
+
if submit and input:
|
20 |
+
|
21 |
+
|
22 |
+
response=chat.send_message(input,stream=True)
|
23 |
+
st.session_state['chat_history'].append(("You", input))
|
24 |
+
st.subheader("The Response is")
|
25 |
+
for chunk in response:
|
26 |
+
st.write(chunk.text)
|
27 |
+
st.session_state['chat_history'].append(("Bot", chunk.text))
|
28 |
+
|
29 |
+
|
30 |
+
st.subheader("The Chat History is")
|
31 |
+
|
32 |
+
for role, text in st.session_state['chat_history']:
|
33 |
st.write(f"{role}: {text}")
|