Update app.py

app.py

@@ -2,8 +2,6 @@ import streamlit as st
 import google.generativeai as genai
 import os
 from dotenv import load_dotenv
-from mimetypes import guess_type as guessmime
-from io import BytesIO
 
 load_dotenv()
 # Configure the API key
@@ -16,36 +14,15 @@ 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":
         return "assistant"
     else:
         return role
-    
-def create_media_part(data, mimetype,filename=None):
-        """Creates a media part for the GenerativeModel.
-
-        Args:
-            data: The image data as bytes.
-            mimetype: The mimetype of the image.
-            filename: optional filename for the image part
-
-        Returns:
-            A Part object representing the image.
-        """
-        # The API expects raw bytes so avoid having it base64 encode the input
-        blob = genai.types.Blob(mimetype, genai.types.bytes_to_data(data, mime_type=mimetype))
-        return genai.types.Part(filename=filename, inline_data=blob)
 
 # Add a Gemini Chat history object to Streamlit session state
 if "chat" not in st.session_state:
@@ -54,41 +31,54 @@ if "chat" not in st.session_state:
 # Display Form Title
 st.title("Mariam AI!")
 
-# Handle file uploads outside the input loop
-uploaded_file = st.file_uploader("Choose a file", type=["png", "jpg", "jpeg", "mp3", "wav", "mp4", "avi"])
+# File upload section
+uploaded_file = st.file_uploader("Télécharger un fichier (image/document)", type=['jpg', 'jpeg', 'png', 'pdf', 'txt'])
 
 # 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)):
         st.markdown(message.parts[0].text)
 
+# Function to handle file upload with Gemini
+def process_uploaded_file(file):
+    if file is not None:
+        # Save the uploaded file temporarily
+        with open(os.path.join("temp", file.name), "wb") as f:
+            f.write(file.getbuffer())
+        
+        # Upload the file to Gemini
+        try:
+            gemini_file = genai.upload_file(os.path.join("temp", file.name))
+            return gemini_file
+        except Exception as e:
+            st.error(f"Erreur lors du téléchargement du fichier : {e}")
+            return None
+
 # Accept user's next message, add to context, resubmit context to Gemini
 if prompt := st.chat_input("Hey?"):
+    # Process any uploaded file
+    uploaded_gemini_file = None
+    if uploaded_file:
+        uploaded_gemini_file = process_uploaded_file(uploaded_file)
+    
     # Display user's last message
     st.chat_message("user").markdown(prompt)
 
-    if uploaded_file is not None:
-        # Display the uploaded file
-        if uploaded_file.type.startswith('image'):
-            st.image(uploaded_file, caption="Uploaded Image.", use_column_width=True)
-        elif uploaded_file.type.startswith('audio'):
-            st.audio(uploaded_file, format=uploaded_file.type)
-        elif uploaded_file.type.startswith('video'):
-            st.video(uploaded_file, format=uploaded_file.type)
+    # Send user entry to Gemini with optional file
+    try:
+        if uploaded_gemini_file:
+            # If a file is uploaded, include it in the context
+            response = st.session_state.chat.send_message([uploaded_gemini_file, "\n\n", prompt])
+        else:
+            # Normal text-only conversation
+            response = st.session_state.chat.send_message(prompt)
+
+        # Display last response
+        with st.chat_message("assistant"):
+            st.markdown(response.text)
     
-        file_bytes = uploaded_file.getvalue()
-        mime_type = guessmime(uploaded_file.name)[0]
-        media_part = create_media_part(file_bytes, mime_type,filename=uploaded_file.name)
-        
-        # Combine prompt and uploaded file for Gemini
-        parts = [media_part, "\n\n", prompt]
-        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)
+    except Exception as e:
+        st.error(f"Erreur lors de l'envoi du message : {e}")
 
-    # Display last
-    with st.chat_message("assistant"):
-        st.markdown(response.text)
+# Create temp directory if it doesn't exist
+os.makedirs("temp", exist_ok=True)