Update interface.py

interface.py

@@ -32,7 +32,7 @@ class ChatInterface:
     Supports both regular image files and DICOM medical imaging files.
     """
 
-    def __init__(self, agent, tools_dict, session_details):
+    def __init__(self, agent, tools_dict):
         """
         Initialize the chat interface.
 
@@ -48,7 +48,6 @@ class ChatInterface:
         # Separate storage for original and display paths
         self.original_file_path = None  # For LLM (.dcm or other)
         self.display_file_path = None  # For UI (always viewable format)
-        self.session_details = session_details
 
     def handle_upload(self, file_path: str) -> str:
         """
@@ -104,7 +103,7 @@ class ChatInterface:
         return history, gr.Textbox(value=message, interactive=False)
 
     async def process_message(
-        self, message: str, display_image: Optional[str], chat_history: List[ChatMessage]
+        self, message: str, display_image: Optional[str], session_details: dict, chat_history: List[ChatMessage]
     ) -> AsyncGenerator[Tuple[List[ChatMessage], Optional[str], str], None]:
         """
         Process a message and generate responses.
@@ -202,7 +201,7 @@ class ChatInterface:
             yield chat_history, self.display_file_path
 
         finally:
-            store_chat_history(self.session_details['username'], self.session_details['session_id'], chat_history)
+            store_chat_history(session_details['username'], session_details['session_id'], chat_history)
 
 def store_chat_history(username, session_id, chat_history):
     """
@@ -415,7 +414,7 @@ def create_demo(agent, tools_dict):
         bot_msg = chat_msg.then(
             interface.process_message,
             inputs=[txt, image_display, chatbot],
-            outputs=[chatbot, image_display, txt],
+            outputs=[chatbot, image_display, session_details, txt],
         )
         bot_msg.then(lambda: gr.Textbox(interactive=True), None, [txt])
 
@@ -426,7 +425,7 @@ def create_demo(agent, tools_dict):
         ).then(
             interface.process_message,
             inputs=[txt, image_display, chatbot],
-            outputs=[chatbot, image_display, txt],
+            outputs=[chatbot, image_display, session_details, txt],
         ).then(lambda: gr.Textbox(interactive=True), None, [txt])
         
         analyze2_btn.click(
@@ -436,7 +435,7 @@ def create_demo(agent, tools_dict):
         ).then(
             interface.process_message,
             inputs=[txt, image_display, chatbot],
-            outputs=[chatbot, image_display, txt],
+            outputs=[chatbot, image_display, session_details, txt],
         ).then(lambda: gr.Textbox(interactive=True), None, [txt])
 
         segment_btn.click(
@@ -445,7 +444,7 @@ def create_demo(agent, tools_dict):
             interface.add_message, inputs=[txt, image_display, chatbot], outputs=[chatbot, txt] 
         ).then(
             interface.process_message,
-            inputs=[txt, image_display, chatbot],
+            inputs=[txt, image_display, session_details, chatbot],
             outputs=[chatbot, image_display, txt],
         ).then(lambda: gr.Textbox(interactive=True), None, [txt])