Update app.py

app.py

@@ -5,42 +5,27 @@ from PIL import Image
 import io
 import os
 
-class ChatBot:
-    def __init__(self, image_base64, api_key):
-        self.image_base64 = image_base64
-        self.api_key = api_key
-        self.contents = [
+# Function to make an API call to Google's Gemini API
+def call_gemini_api(image_base64, api_key, prompt="What is this picture?"):
+    headers = {
+        "Content-Type": "application/json",
+    }
+    data = {
+        "contents": [
             {
                 "parts": [
-                    {"text": "You are a helpful assistant."},
+                    {"text": prompt},
                     {"inline_data": {"mime_type": "image/jpeg", "data": image_base64}},
                 ]
             }
         ]
-
-    def generate_response(self, prompt: str) -> str:
-        self.contents.append(
-            {
-                "parts": [
-                    {"text": prompt},
-                    {"inline_data": {"mime_type": "image/jpeg", "data": self.image_base64}},
-                ]
-            }
-        )
-        headers = {
-            "Content-Type": "application/json",
-        }
-        data = {"contents": self.contents}
-        response = requests.post(
-            f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro-vision:generateContent?key={self.api_key}",
-            headers=headers,
-            json=data,
-        )
-        response_json = response.json()
-        return response_json
-
-    def get_history(self) -> list:
-        return self.contents
+    }
+    response = requests.post(
+        f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro-vision:generateContent?key={api_key}",
+        headers=headers,
+        json=data,
+    )
+    return response.json()
 
 # Streamlit app
 st.set_page_config(layout="wide")
@@ -58,8 +43,6 @@ if uploaded_file is not None:
     api_key = os.environ["GEMINI_API_KEY"] # st.sidebar.text_input("Enter your API key", type="password")
 
     if api_key:
-        chatbot = ChatBot(image_base64, api_key)
-        
         st.header("Chat with the Bot")
         
         if 'conversation' not in st.session_state:
@@ -68,9 +51,8 @@ if uploaded_file is not None:
         user_input = st.text_input("Ask a question about the image:")
         
         if user_input:
-            json_response = chatbot.generate_response(user_input)
-            st.write(json_response)
-            # response = json_response["candidates"][0]["content"]["parts"][0]["text"]
+            json_response = call_gemini_api(image_base64, api_key, user_input)
+            response = json_response["candidates"][0]["content"]["parts"][0]["text"]
             st.session_state.conversation.append({"user": user_input, "bot": response})
         
         if st.session_state.conversation: