Update app.py

app.py

@@ -40,37 +40,48 @@ def send_message():
     user_input = st.session_state.user_input
     uploaded_files = st.session_state.uploaded_files
     
-    if user_input or uploaded_files:
-        # Save user input to the chat history
-        if user_input:
-            st.session_state['chat_history'].append({"role": "user", "parts": [{"text": user_input}]})
-
-        # Process uploaded images
-        image_prompts = []
-        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_prompt = {"role": "user", "parts": [{"mime_type": uploaded_file.type, "data": image_base64}]}
-                image_prompts.append(image_prompt)
-            st.session_state['chat_history'].extend(image_prompts)
-
-        # Choose the appropriate model based on the input type
-        model_name = 'gemini-pro-vision' if uploaded_files else 'gemini-pro'
+    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}]})
+
+    # 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=model_name,
+            model_name='gemini-pro',
             generation_config=generation_config,
             safety_settings=safety_settings
         )
-
-        # Generate the response
         response = model.generate_content(st.session_state['chat_history'])
         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:
+        model = genai.GenerativeModel(
+            model_name='gemini-pro-vision',
+            generation_config=generation_config,
+            safety_settings=safety_settings
+        )
+        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.file_uploader_key = str(uuid.uuid4())
+    # 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.file_uploader_key = str(uuid.uuid4())
 
 # Multiline text input for the user to send messages
 user_input = st.text_area("Enter your message here:", key="user_input", value="")