Update app.py

app.py

@@ -3,9 +3,11 @@ import google.generativeai as genai
 import os
 from dotenv import load_dotenv
 from PIL import Image
+import io
 import mimetypes
 
 load_dotenv()
+
 # Configure the API key
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
@@ -16,15 +18,11 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
-genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-
-model = genai.GenerativeModel('gemini-1.5-flash',safety_settings=safety_settings,
-            system_instruction="Tu es un assistant intelligent. ton but est d'assister au mieux que tu peux. tu as été créé par Aenir et tu t'appelles Mariam")
-
-
-
-# Function to get response from the model
-# Gemini uses 'model' for assistant; Streamlit uses 'assistant'
+model = genai.GenerativeModel(
+    'gemini-1.5-flash',
+    safety_settings=safety_settings,
+    system_instruction="Tu es un assistant intelligent. ton but est d'assister au mieux que tu peux. tu as été créé par Aenir et tu t'appelles Mariam"
+)
 
 def role_to_streamlit(role):
     if role == "model":
@@ -42,63 +40,52 @@ st.title("Mariam AI!")
 # Display chat messages from history above current input box
 for message in st.session_state.chat.history:
     with st.chat_message(role_to_streamlit(message.role)):
-        # Check if the message part is text or a file part
         for part in message.parts:
-            if part.HasField("text"):
+            if part.text:  # Check for text content
                 st.markdown(part.text)
-            elif part.HasField("file_data"):
-                # Handle file display (e.g., image)
+            elif part.file_data:  # Check for file data
                 try:
                     # Infer MIME type if not provided
-                    if not part.file_data.mime_type:
+                    mime_type = part.file_data.mime_type
+                    if not mime_type:
                         mime_type = mimetypes.guess_type(part.file_data.file_name)[0]
-                    else:
-                        mime_type = part.file_data.mime_type
+
                     if mime_type and mime_type.startswith("image/"):
-                        image_data = part.file_data.data  # Access the image data directly
-                        image = Image.open(io.BytesIO(image_data)) # Open the image using PIL
+                        image = Image.open(io.BytesIO(part.file_data.data))
                         st.image(image)
                     else:
-                        st.write(f"File: {part.file_data.file_name} (MIME type: {part.file_data.mime_type})")
+                        st.write(f"File: {part.file_data.file_name} (MIME type: {mime_type})")
                 except Exception as e:
                     st.error(f"Error displaying file: {e}")
 
-# Accept user's next message, add to context, resubmit context to Gemini
+# Accept user's next message and file uploads
 if prompt := st.chat_input("Hey?"):
-    # Display user's last message
-    st.chat_message("user").markdown(prompt)
-
-    # Handle file uploads
-    uploaded_file = st.file_uploader("Choose a file", type=["jpg", "jpeg", "png", "pdf"]) # Add more type if needed
+    uploaded_file = st.file_uploader("Choose a file", type=["jpg", "jpeg", "png", "pdf"])
 
-    if uploaded_file is not None:
-        # To read file as bytes:
+    parts = [prompt]
+    if uploaded_file:
         bytes_data = uploaded_file.getvalue()
+        parts.append({
+            "file_data": {
+                "mime_type": uploaded_file.type,
+                "file_name": uploaded_file.name,
+                "data": bytes_data
+            }
+        })
 
-        # Display the uploaded file (if it's an image)
+        # Display the uploaded image
         if uploaded_file.type.startswith("image/"):
             image = Image.open(uploaded_file)
-            st.image(image, caption=f"Uploaded Image: {uploaded_file.name}")
+            with st.chat_message("user"):
+                st.image(image, caption=f"Uploaded Image: {uploaded_file.name}")
 
-        # Construct the message parts, including the uploaded file
-        parts = [
-            prompt,
-            {
-                "file_data": {
-                    "mime_type": uploaded_file.type,
-                    "file_name": uploaded_file.name,
-                    "data": bytes_data
-                }
-            }
-        ]
+    # Display user's message
+    with st.chat_message("user"):
+        st.markdown(prompt)
 
-        # Send the message with file to Gemini
-        response = st.session_state.chat.send_message(parts)
-    
-    else:
-      # Send user entry to Gemini and read the response
-      response = st.session_state.chat.send_message(prompt)
+    # Send message to Gemini
+    response = st.session_state.chat.send_message(parts)
 
-    # Display last
+    # Display Gemini's response
     with st.chat_message("assistant"):
         st.markdown(response.text)