Upload chatapp.py

chatapp.py

@@ -0,0 +1,50 @@
+import pathlib
+import textwrap
+import streamlit as st
+import google.generativeai as genai
+from dotenv import load_dotenv
+from IPython.display import Markdown
+from IPython.display import display
+import os
+
+load_dotenv()
+genai.configure(api_key="AIzaSyA_XykKxC4aSi0af9VH5uP2eQlp9Nh25Ds")
+#model = genai.GenerativeModel(model_name="models/gemini-pro")
+model = genai.GenerativeModel("gemini-pro")
+chat = model.start_chat(history=[])
+
+def to_markdown(text):
+    text = text.replace('•', ' *')
+    return Markdown(textwrap.indent(text,'> ',predicate=lambda _:True))
+
+def get_gemini_response(question):
+    #response  = model.generate_content(question)
+    
+    response = chat.send_message(question,stream=True)
+    #to_markdown(response.text)
+    #print('Response:',response.text)
+    #response = to_markdown(response.text)
+    return response
+
+st.title("Google Powered Chatbot")
+
+if 'chat_history' not in st.session_state:
+    st.session_state['chat_history'] = []
+
+user_input = st.text_input("Input:",key='input')
+submit_button = st.button("Ask Question")
+
+if submit_button and user_input:
+    if user_input:
+        bot_response = get_gemini_response(user_input)
+        st.session_state['chat_history'].append(("You",user_input))
+        for chunk in bot_response:
+            st.write(chunk.text)
+            st.session_state['chat_history'].append(("Bot",chunk.text))
+        #st.write("Bot:", bot_response)
+    else:
+        st.warning("Please enter something.")
+# Function to interact with OpenAI API and get response
+for role,text in st.session_state['chat_history']:
+    st.write(f"{role}:{text}")
+