Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 from PIL import Image
-from function import bounding_box, captioning_query
+from function import bounding_box
 from tempfile import NamedTemporaryFile
 import os
 from function import ImageCaptionTools, ObjectDetectionTool
@@ -8,10 +8,25 @@ from langchain.agents import initialize_agent, AgentType
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.memory import ConversationBufferWindowMemory
 from htmlTemplate import css, bot_template, user_template
+import random
+
+DIR = './temp'
+if not os.path.exists(DIR):
+    os.mkdir(DIR)
+
+if "image_processed" not in st.session_state:
+    DIR_PATH = os.path.join(DIR, str(random.randint(1,999999999)))
+    st.session_state.dirpath = DIR_PATH
+    if not os.path.exists(DIR_PATH):
+        os.mkdir(DIR_PATH)
+
+def delete_temp_files():
+    for filename in os.listdir(st.session_state.dirpath):
+        file_path = os.path.join(st.session_state.dirpath, filename)
+        if os.path.isfile(file_path):
+            os.unlink(file_path)
+
 
-DIR_PATH = './temp'
-if not os.path.exists(DIR_PATH):
-    os.mkdir(DIR_PATH)
 
 # initialize Agent
 def agent_init():
@@ -30,11 +45,7 @@ def agent_init():
     )
     return agents
 
-def delete_temp_files():
-    for filename in os.listdir(DIR_PATH):
-        file_path = os.path.join(DIR_PATH, filename)
-        if os.path.isfile(file_path):
-            os.unlink(file_path)
+
 
 def main():
     st.set_page_config(
@@ -46,16 +57,18 @@ def main():
     st.title("Chat with an Image 🖼️")
     agent = agent_init()
 
+    # Check if the page has been reloaded
+    if 'reloaded' not in st.session_state:
+        st.session_state.reloaded = False
+    else:
+        st.session_state.reloaded = True
+
     if "image_processed" not in st.session_state:
         st.session_state.image_processed = None
 
     if "result_bounding" not in st.session_state:
         st.session_state.result_bounding = None
 
-    # Delete temp files when session state changes
-    if st.session_state.image_processed is None:
-        delete_temp_files()
-
     # image_path = 'documentation\photo_1.jpg'
 
     col1, col2 = st.columns([1, 1])
@@ -71,7 +84,7 @@ def main():
         click_process = st.button("Process Image", disabled=not image_upload)
         if click_process:
             delete_temp_files()
-            with NamedTemporaryFile(dir=DIR_PATH, delete=False) as f:
+            with NamedTemporaryFile(dir=st.session_state.dirpath, delete=False) as f:
                 f.write(image_upload.getbuffer())
                 st.session_state.image_path = f.name
                 st.session_state.image_processed = True
@@ -93,7 +106,7 @@ def main():
         click_ask = st.button("Ask Question", disabled=not st.session_state.image_processed)
         if click_ask:
             st.write(user_template.replace("{{MSG}}", user_question), unsafe_allow_html=True)
-            with st.spinner("AI Searching for Answer🔎"):
+            with st.spinner("Doraemon Searching for Answer🔎"):
                 chat_history = agent.invoke({"input": f"{user_question}, this is the image path: {st.session_state.image_path}"})
                 response = chat_history['output']
                 st.write(bot_template.replace("{{MSG}}", response), unsafe_allow_html=True)