Update app.py

app.py

@@ -4,20 +4,18 @@ import io
 import base64
 import uuid
 
-# Placeholder import - Replace with the actual import once you have the correct library
+# Assuming google.generativeai is the correct import based on your description
 import google.generativeai as genai
 
-# Set up your API key securely, don't hardcode it in production
-api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
+# Configure the API key (should be set as an environment variable or secure storage in production)
+api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"  # Replace with your actual API key
 genai.configure(api_key=api_key)
 
-# Placeholder for the generative model configuration
 generation_config = genai.GenerationConfig(
     temperature=0.9,
     max_output_tokens=3000
 )
 
-# Placeholder for safety settings
 safety_settings = []
 
 # Initialize session state for chat history and file uploader key
@@ -37,6 +35,11 @@ def get_image_base64(image):
     img_str = base64.b64encode(buffered.getvalue()).decode()
     return img_str
 
+# Function to clear conversation
+def clear_conversation():
+    st.session_state['chat_history'] = []
+    st.session_state['file_uploader_key'] = str(uuid.uuid4())
+
 # Function to send message and clear input
 def send_message():
     user_input = st.session_state.user_input
@@ -44,48 +47,47 @@ def send_message():
 
     # Process text input for multi-turn conversation
     if user_input:
-        text_prompt = {"role": "user", "parts": [{"text": user_input}]}
-        st.session_state['chat_history'].append(text_prompt)
-        # Actual API call for generating a text response from the Gemini Pro model
-        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_text = response.text if hasattr(response, "text") else "No response text found."
-        st.session_state['chat_history'].append({"role": "model", "parts": [{"text": response_text}]})
+        st.session_state['chat_history'].append({"role": "user", "parts": [{"text": user_input}]})
 
-    # Process uploaded images for single-turn conversation
+    # Check if images are uploaded
     if uploaded_files:
-        image_prompts = []
-        for uploaded_file in uploaded_files:
-            image = Image.open(uploaded_file).convert("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)
-        # Actual API call for generating a response from the Gemini Pro Vision model
-        model = genai.GenerativeModel(
+        # Prepare image prompts for single-turn conversation
+        image_prompts = [{
+            "role": "user",
+            "parts": [{"mime_type": uploaded_file.type, "data": get_image_base64(Image.open(uploaded_file))}]
+        } for uploaded_file in uploaded_files]
+        
+        # Use Gemini Pro Vision model for image-based interaction
+        vision_model = genai.GenerativeModel(
             model_name='gemini-pro-vision',
             generation_config=generation_config,
             safety_settings=safety_settings
         )
-        response = model.generate_content(image_prompts)
+        response = vision_model.generate_content(image_prompts)
         response_text = response.text if hasattr(response, "text") else "No response text found."
+        
+        # Append images and response to chat history
         for prompt in image_prompts:
             st.session_state['chat_history'].append(prompt)
         st.session_state['chat_history'].append({"role": "model", "parts": [{"text": response_text}]})
+        
+    # If no images are uploaded, use Gemini Pro model for text-based interaction
+    elif user_input:
+        text_model = genai.GenerativeModel(
+            model_name='gemini-pro',
+            generation_config=generation_config,
+            safety_settings=safety_settings
+        )
+        response = text_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}]})
 
-    # Clear input and uploaded files after processing
+    # 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 = []
     st.session_state.file_uploader_key = str(uuid.uuid4())
 
-# Function to clear conversation
-def clear_conversation():
-    st.session_state['chat_history'] = []
-
-# Input field for text messages
+# Multiline text input for the user to send messages
 user_input = st.text_area("Enter your message here:", key="user_input")
 
 # File uploader for images