Update app.py

app.py

@@ -18,7 +18,7 @@ generation_config = genai.GenerationConfig(
 
 safety_settings = []
 
-# Initialize session state for chat history and file uploader key
+# Initialize session state for chat history
 if 'chat_history' not in st.session_state:
     st.session_state['chat_history'] = []
 if 'file_uploader_key' not in st.session_state:
@@ -39,35 +39,31 @@ def get_image_base64(image):
 def send_message():
     user_input = st.session_state.user_input
     uploaded_files = st.session_state.uploaded_files
-    
-    text_prompts = []
-    image_prompts = []
-    
-    # Process text input for multi-turn conversation
+
     if user_input:
-        text_prompts.append({"role": "user", "parts": [{"text": user_input}]})
-        st.session_state['chat_history'].append({"role": "user", "parts": [{"text": user_input}]})
+        # Send text to the gemini-pro model for text-based conversation
+        text_prompt = {"role": "user", "parts": [{"text": user_input}]}
+        st.session_state['chat_history'].append(text_prompt)
 
-    # Process uploaded images for single-turn conversation
-    if uploaded_files:
-        for uploaded_file in uploaded_files:
-            image = Image.open(uploaded_file).convert("RGB")  # Ensure image is in RGB
-            image_base64 = get_image_base64(image)
-            image_prompts.append({"role": "user", "parts": [{"mime_type": uploaded_file.type, "data": image_base64}]})
-    
-    # Generate text response if text input is provided
-    if text_prompts:
         model = genai.GenerativeModel(
             model_name='gemini-pro',
             generation_config=generation_config,
             safety_settings=safety_settings
         )
-        response = model.generate_content(st.session_state['chat_history'])
+        response = model.generate_content([text_prompt])
         response_text = response.text if hasattr(response, "text") else "No response text found."
         st.session_state['chat_history'].append({"role": "model", "parts": [{"text": response_text}]})
-    
-    # Generate image response if images are uploaded
-    if image_prompts:
+
+    if uploaded_files:
+        # Send images to the gemini-pro-vision model for image-based conversation
+        image_prompts = []
+        for uploaded_file in uploaded_files:
+            image = Image.open(uploaded_file).convert("RGB")  # Ensure image is in RGB
+            image_base64 = get_image_base64(image)
+            image_prompt = {"role": "user", "parts": [{"mime_type": uploaded_file.type, "data": image_base64}]}
+            image_prompts.append(image_prompt)
+            st.session_state['chat_history'].append(image_prompt)
+
         model = genai.GenerativeModel(
             model_name='gemini-pro-vision',
             generation_config=generation_config,
@@ -75,16 +71,18 @@ def send_message():
         )
         response = model.generate_content(image_prompts)
         response_text = response.text if hasattr(response, "text") else "No response text found."
-        for prompt in image_prompts:
-            st.session_state['chat_history'].append(prompt)  # Append images to history
         st.session_state['chat_history'].append({"role": "model", "parts": [{"text": response_text}]})
 
     # Clear the user input and generate a new key for the file uploader widget to reset it
     st.session_state.user_input = ''
+    st.session_state.uploaded_files = None
     st.session_state.file_uploader_key = str(uuid.uuid4())
 
+def clear_conversation():
+    st.session_state.chat_history = []
+
 # Multiline text input for the user to send messages
-user_input = st.text_area("Enter your message here:", key="user_input", value="")
+user_input = st.text_area("Enter your message here:", key="user_input")
 
 # File uploader for images
 uploaded_files = st.file_uploader(
@@ -97,6 +95,9 @@ uploaded_files = st.file_uploader(
 # Button to send the message
 send_button = st.button("Send", on_click=send_message)
 
+# Button to clear the conversation
+clear_button = st.button("Clear Conversation", on_click=clear_conversation)
+
 # Display the chat history
 for entry in st.session_state['chat_history']:
     role = entry["role"]
@@ -104,7 +105,10 @@ for entry in st.session_state['chat_history']:
     if 'text' in parts:
         st.markdown(f"**{role.title()}**: {parts['text']}")
     elif 'data' in parts:
+        # Decode the base64 image data
+        base64_data = parts['data']
         st.markdown(f"**{role.title()}**: (Image)")
+        st.image(base64_data, caption=f"Image from {role}", use_column_width=True)
 
 # Ensure the file_uploader widget state is tied to the randomly generated key
 st.session_state.uploaded_files = uploaded_files