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):
+    def __init__(self, agent, tools_dict, session_details):
         """
         Initialize the chat interface.
 
@@ -48,6 +48,7 @@ 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:
         """
@@ -103,7 +104,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], session_details: dict
+        self, message: str, display_image: Optional[str], chat_history: List[ChatMessage]
     ) -> AsyncGenerator[Tuple[List[ChatMessage], Optional[str], str], None]:
         """
         Process a message and generate responses.
@@ -201,7 +202,7 @@ class ChatInterface:
             yield chat_history, self.display_file_path
 
         finally:
-            store_chat_history(session_details['username'], session_details['session_id'], chat_history)
+            store_chat_history(self.session_details['username'], self.session_details['session_id'], chat_history)
 
 def store_chat_history(username, session_id, chat_history):
     """
@@ -413,7 +414,7 @@ def create_demo(agent, tools_dict):
         )
         bot_msg = chat_msg.then(
             interface.process_message,
-            inputs=[txt, image_display, chatbot, session_details],
+            inputs=[txt, image_display, chatbot],
             outputs=[chatbot, image_display, txt],
         )
         bot_msg.then(lambda: gr.Textbox(interactive=True), None, [txt])
@@ -424,7 +425,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, session_details],
+            inputs=[txt, image_display, chatbot],
             outputs=[chatbot, image_display, txt],
         ).then(lambda: gr.Textbox(interactive=True), None, [txt])
         
@@ -434,7 +435,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, session_details],
+            inputs=[txt, image_display, chatbot],
             outputs=[chatbot, image_display, txt],
         ).then(lambda: gr.Textbox(interactive=True), None, [txt])
 
@@ -444,7 +445,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, session_details],
+            inputs=[txt, image_display, chatbot],
             outputs=[chatbot, image_display, txt],
         ).then(lambda: gr.Textbox(interactive=True), None, [txt])