Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
-from pathlib import Path
 from PIL import Image
-from libs.gemini_vision import GeminiVision
+from google.generativeai import GenerativeModel  # Assuming the actual module from Gemini Vision API
+from pathlib import Path
 
 # Initialize session state
 def init_session_state():
@@ -13,24 +13,20 @@ def init_session_state():
         st.session_state['top_k'] = 32
     if 'top_p' not in st.session_state:
         st.session_state['top_p'] = 1.0
-    if 'gemini_vision' not in st.session_state:
-        st.session_state['gemini_vision'] = None
-
-# Exception handling decorator
-def exception_handler(func):
-    def wrapper(*args, **kwargs):
-        try:
-            return func(*args, **kwargs)
-        except Exception as exception:
-            st.session_state.logger.error(f"An error occurred in {func.__name__}: {exception}")
-            st.error(f"An error occurred: {exception}")
-            st.session_state.logger.error(traceback.format_exc())
-            st.stop()
-    return wrapper
+    if 'gemini_model' not in st.session_state:
+        st.session_state['gemini_model'] = None
+    if 'uploaded_image' not in st.session_state:
+        st.session_state['uploaded_image'] = None
+    if 'image_description' not in st.session_state:
+        st.session_state['image_description'] = ''
 
 # Streamlit App
-def streamlit_app():
-    st.title("Gemini Vision - Image Descriptions")
+def streamlit_app():    
+    # Google Logo and Title
+    st.write('<div style="display: flex; flex-direction: row; align-items: center; justify-content: center;"><a style="margin-right: 10px;" href="https://www.google.com" target="_blank"><img src="https://img.icons8.com/color/32/000000/google-logo.png"/></a><h1 style="margin-left: 10px;">Google - Gemini Vision</h1></div>', unsafe_allow_html=True)
+
+    # Display support
+    display_support()
 
     # Display the Gemini Sidebar settings
     with st.sidebar.title("Gemini Settings"):
@@ -39,28 +35,51 @@ def streamlit_app():
         st.session_state.top_k = st.sidebar.number_input("Top K", value=32)
         st.session_state.top_p = st.sidebar.slider("Top P", 0.0, 1.0, 1.0)
 
-        if st.button("Apply Settings"):
-            st.session_state.gemini_vision = GeminiVision(
-                st.session_state.api_key,
-                st.session_state.temperature,
-                st.session_state.top_p,
-                st.session_state.top_k
-            )
-            st.success("Settings applied successfully!")
+        if (st.session_state.api_key is not None and st.session_state.api_key != '') \
+            and (st.session_state.temperature is not None and st.session_state.temperature != '') \
+            and (st.session_state.top_k is not None and st.session_state.top_k != '') \
+            and (st.session_state.top_p is not None and st.session_state.top_p != ''):
+            st.toast("Settings updated successfully!", icon="👍")
+        else:
+            st.toast("Please enter all the settings.\nAPI Key is required", icon="❌")
+            raise ValueError("Please enter all values the settings.\nAPI Key is required")
+
+    # Initialize services once
+    if st.session_state.gemini_model is None:
+        st.session_state.gemini_model = GenerativeModel(model_name="gemini-pro-vision",
+                                                         api_key=st.session_state['api_key'],
+                                                         temperature=st.session_state['temperature'],
+                                                         top_p=st.session_state['top_p'],
+                                                         top_k=st.session_state['top_k'])
 
-    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+    uploaded_image = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
 
-    if uploaded_file is not None:
+    if uploaded_image:
         # Display the uploaded image
-        st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
+        st.image(uploaded_image, caption="Uploaded Image", use_column_width=True)
+        
+        # Store the uploaded image in session state
+        st.session_state['uploaded_image'] = uploaded_image
+
+        if st.button("Generate Image Description"):
+            try:
+                # Process the uploaded image using Gemini Vision API
+                image_description = st.session_state.gemini_model.generate_content(contents=[uploaded_image])
+                st.session_state['image_description'] = image_description.text
+                st.success(f"Image description generated: {image_description.text}")
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
 
-        if st.session_state.gemini_vision is not None:
-            # Generate description using Gemini Vision API
-            image = Image.open(uploaded_file)
-            description = st.session_state.gemini_vision.generate_content(contents=[image])
-            st.success(f"Gemini Vision Description: {description.text}")
+    # Display the generated image description
+    if st.session_state['image_description']:
+        st.code(f"Image Description: {st.session_state['image_description']}", language="plaintext")
 
-# Main
 if __name__ == "__main__":
-    init_session_state()
-    streamlit_app()
+    try:
+        init_session_state()
+        streamlit_app()
+    except Exception as exception:
+        import traceback
+        st.session_state.logger.error(f"An error occurred: {exception}")
+        st.session_state.logger.error(traceback.format_exc())
+        st.error(f"An error occurred: {exception}")