gyanbardhan123 commited on
Commit
b351768
·
verified ·
1 Parent(s): 66f22c4

Update Chatbot.py

Browse files
Files changed (1) hide show
  1. 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.0-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}")
 
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}")