File size: 3,075 Bytes
7103623 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
import streamlit as st
import google.generativeai as genai
from PIL import Image
import io
st.set_page_config(page_title="Gemini AI Chat", layout="wide")
st.title("🤖 Gemini AI Chat Interface")
st.markdown("Chat with Google's Gemini AI models. Supports both text and image inputs.")
# Sidebar for settings
with st.sidebar:
api_key = st.text_input("Enter Google AI API Key", type="password")
model = st.selectbox(
"Select Model",
[
"gemini-1.5-pro",
"gemini-1.5-pro-vision-latest",
"gemini-1.0-pro",
"gemini-1.0-pro-vision-latest"
]
)
temperature = st.slider("Temperature", 0.0, 1.0, 0.7)
max_tokens = st.slider("Max Tokens", 1, 2048, 1000)
system_prompt = st.text_area("System Prompt (Optional)")
# Initialize session state for chat history
if "messages" not in st.session_state:
st.session_state.messages = []
# Display chat history
for message in st.session_state.messages:
with st.chat_message(message["role"]):
st.markdown(message["content"])
# File uploader for images
uploaded_file = st.file_uploader("Upload an image (optional)", type=["jpg", "jpeg", "png"])
uploaded_image = None
if uploaded_file is not None:
uploaded_image = Image.open(uploaded_file).convert('RGB')
st.image(uploaded_image, caption="Uploaded Image", use_column_width=True)
# Chat input
user_input = st.chat_input("Type your message here...")
if user_input and api_key:
try:
# Configure the API
genai.configure(api_key=api_key)
# Add user message to chat history
st.session_state.messages.append({"role": "user", "content": user_input})
with st.chat_message("user"):
st.markdown(user_input)
# Prepare the model and content
model_instance = genai.GenerativeModel(model_name=model)
content = [{"text": user_input}]
if uploaded_image:
content.append({"inline_data": {"mime_type": "image/jpeg", "data": uploaded_image}})
# Generate response
response = model_instance.generate_content(
content,
generation_config=genai.types.GenerationConfig(
temperature=temperature,
max_output_tokens=max_tokens
)
)
# Display assistant response
with st.chat_message("assistant"):
st.markdown(response.text)
# Add assistant response to chat history
st.session_state.messages.append({"role": "assistant", "content": response.text})
except Exception as e:
st.error(f"Error: {str(e)}")
elif not api_key and user_input:
st.warning("Please enter your API key in the sidebar first.")
# Instructions in the sidebar
with st.sidebar:
st.markdown("""
## 📝 Instructions:
1. Enter your Google AI API key
2. Select a model
3. Adjust temperature and max tokens if needed
4. Optional: Set a system prompt
5. Upload an image (optional)
6. Type your message and press Enter
""") |