Kushwanth Chowday Kandala
update app.py
ff0e82b
raw
history blame
1.58 kB
import streamlit as st
import google.generativeai as genai
import os
from streamlit_chat import message
gemini_api_key = os.getenv("GEMINI_API_KEY")
genai.configure(api_key = gemini_api_key)
model = genai.GenerativeModel('gemini-pro')
# prompt = st.chat_input("Say something")
# if prompt:
# st.write(f"User has sent the following prompt: {prompt}")
# else:
# prompt = "who are you?"
# response = model.generate_content(prompt)
# message = st.chat_message("ai")
# message.write(response.text)
def chat_actions():
st.session_state["chat_history"].append(
{"role": "user", "content": st.session_state["chat_input"]},
)
response = model.generate_content(st.session_state["chat_input"])
st.session_state["chat_history"].append(
{
"role": "assistant",
"content": response.text,
}, # This can be replaced with your chat response logic
)
if "chat_history" not in st.session_state:
st.session_state["chat_history"] = []
st.chat_input("Enter your message", on_submit=chat_actions, key="chat_input")
for i in st.session_state["chat_history"]:
with st.chat_message(name=i["role"]):
st.write(i["content"])
# img_file_buffer = st.file_uploader('Upload a PNG image', type='png')
import numpy as np
from PIL import Image
img_file_buffer = st.file_uploader('Upload a PNG image', on_change=uploader_callback, type=['png','jpg'])
if img_file_buffer is not None:
image = Image.open(img_file_buffer)
img_array = np.array(image)
def uploader_callback():
print('Uploaded file')