Update app.py

app.py

@@ -3,7 +3,7 @@ from PIL import Image
 import io
 import base64
 
-
+# Placeholder for google.generativeai package
 import google.generativeai as genai
 
 # Configure the API key (should be set as an environment variable or secure storage in production)
@@ -40,7 +40,7 @@ def get_image_base64(image):
     buffered = io.BytesIO()
     image.save(buffered, format="JPEG")
     img_str = base64.b64encode(buffered.getvalue()).decode()
-    return f"data:image/jpeg;base64,{img_str}"
+    return img_str
 
 # When the 'Send' button is clicked, process the input and generate a response
 if st.button("Send"):
@@ -50,38 +50,35 @@ if st.button("Send"):
     
     # Process and save uploaded images to the chat history
     image_parts = []
-    for uploaded_file in uploaded_files:
-        bytes_data = uploaded_file.read()
-        image = Image.open(io.BytesIO(bytes_data))
-        image_base64 = get_image_base64(image)
-        image_parts.append({
-            "mime_type": uploaded_file.type,
-            "data": image_base64
-        })
-        st.session_state.chat_history.append(("user", f"Uploaded image: {uploaded_file.name}"))
+    if uploaded_files:
+        for uploaded_file in uploaded_files:
+            bytes_data = uploaded_file.read()
+            image = Image.open(io.BytesIO(bytes_data))
+            image_base64 = get_image_base64(image)
+            image_parts.append({
+                "mime_type": uploaded_file.type,
+                "data": image_base64
+            })
+            st.session_state.chat_history.append(("user", f"Uploaded image: {uploaded_file.name}"))
 
     # Prepare the prompts for the model
-    prompts = []
-    if user_input.strip():
-        prompts.append({"role": "user", "parts": [{"text": user_input}]})
+    prompts = [{"role": "user", "parts": [{"text": user_input}]}] if user_input.strip() else []
     for image_part in image_parts:
         prompts.append({"role": "user", "parts": [image_part]})
 
+    # Use the appropriate Gemini model based on the inputs
+    model_name = 'gemini-pro-vision' if uploaded_files else 'gemini-pro'
+    model = genai.GenerativeModel(
+        model_name=model_name,
+        generation_config=generation_config,
+        safety_settings=safety_settings
+    )
+
     # Generate the response
-    if image_parts:
-        model = genai.GenerativeModel(
-            model_name='gemini-pro-vision',
-            generation_config=generation_config,
-            safety_settings=safety_settings
-        )
-    else:
-        model = genai.GenerativeModel(
-            model_name='gemini-pro',
-            generation_config=generation_config,
-            safety_settings=safety_settings
-        )
     response = model.generate_content(prompts)
-    response_text = response['text']
+    
+    # Corrected response handling
+    response_text = response.text if hasattr(response, "text") else "No response text found."
 
     # Save the model response to the chat history
     st.session_state.chat_history.append(("model", response_text))