Save

app.py

@@ -0,0 +1,67 @@
+import streamlit as st
+from PIL import Image
+import google.generativeai as genai
+import os
+
+
+MODEL_ID = "gemini-2.0-flash-exp" 
+api_key = os.getenv("GEMINI_API_KEY")
+
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Function to reset chat history
+def reset_chat():
+    st.session_state.messages = []
+
+# Streamlit app
+st.title("Gemini Image Chat")
+
+model_id=MODEL_ID
+genai.configure(api_key=api_key)
+model = genai.GenerativeModel(model_id)
+chat = model.start_chat() 
+
+# File uploader
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+
+if uploaded_file is not None:
+    # Display the uploaded image
+    image = Image.open(uploaded_file)
+    st.image(image, caption="Uploaded Image.", use_container_width=True)
+
+    # Reset chat history when a new image is uploaded
+    reset_chat()
+
+    # Text input for user prompt
+    user_input = st.text_input("Enter your prompt:")
+
+    # Send button
+    if st.button("Send"):
+        if user_input:
+            # Add user message to chat history
+            st.session_state.messages.append({"role": "user", "content": user_input})
+
+            # Display chat history
+            for message in st.session_state.messages:
+                with st.chat_message(message["role"]):
+                    st.markdown(message["content"])
+
+
+            img_file = genai.upload_file(uploaded_file, mime_type="image/jpeg")
+
+            # Send image and prompt to Gemini API
+            response = model.generate_content(
+                [
+                    user_input, 
+                    img_file
+                ]
+            )
+
+            # Add Gemini response to chat history
+            st.session_state.messages.append({"role": "assistant", "content": response.text})
+
+            # Display Gemini response
+            with st.chat_message("assistant"):
+                st.markdown(response.text)

requirements.txt

@@ -0,0 +1,2 @@
+streamlit
+google-generativeai