Update app.py

app.py

@@ -37,14 +37,24 @@ for message in chat_history:
     r, t = message["role"], message["parts"][0]["text"]
     st.markdown(f"**{r.title()}:** {t}")
     
-# Use text_area for multiline input   
-user_input = st.text_area("", height=5)  
+# Use file_uploader for text and image input   
+user_input = st.file_uploader("Upload a text or an image file here", type=["txt", "png", "jpg", "jpeg"])  
 if user_input:
-    chat_history.append({"role": role, "parts": [{"text": user_input}]}) 
+    # Check the file type and create the prompt parts accordingly
+    if user_input.type == "text/plain":
+        prompt_parts = [user_input.getvalue().decode("utf-8")]
+        model_name = "gemini-pro" # Use the text-only model
+    else:
+        prompt_parts = [{
+            "mime_type": user_input.type,
+            "data": user_input.read() 
+        }]
+        model_name = "gemini-pro-vision" # Use the multimodal model
+    
+    chat_history.append({"role": role, "parts": prompt_parts}) 
     if role == "user":
         
         # Model code
-        model_name = "gemini-pro"
         model = genai.GenerativeModel(
             model_name=model_name,   
             generation_config=generation_config,
@@ -74,40 +84,3 @@ for message in chat_history:
     conn.commit() 
 
 conn.close()
-
-# Separate section for image uploading
-st.title("Image Description Generator")
-
-# Change the file_uploader to accept multiple files
-uploaded_files = st.file_uploader("Upload one or more images here", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
-
-# Text input for asking questions about the images
-image_question = st.text_input("Ask something about the images:")
-
-# Check if the user has entered a question
-if image_question:
-    # Create a list of image parts from the uploaded files
-    image_parts = []
-    for uploaded_file in uploaded_files:
-        image_parts.append({
-            "mime_type": uploaded_file.type,
-            "data": uploaded_file.read() 
-        })
-    
-    # Create a prompt parts list with the question and the image parts
-    prompt_parts = [image_question] + image_parts
-
-    # Use the gemini-pro-vision model to generate a response
-    model = genai.GenerativeModel(
-        model_name="gemini-pro-vision", 
-        generation_config=generation_config, 
-        safety_settings=safety_settings 
-    )
-    
-    response = model.generate_content(prompt_parts)
-    st.markdown(f"**Model's answer:** {response.text}")
-
-# Loop through the uploaded files and display them
-for uploaded_file in uploaded_files:
-    # Display the image
-    st.image(uploaded_file)