Update src/insurance_assistants/ui.py

src/insurance_assistants/ui.py

@@ -15,7 +15,7 @@ from smolagents.models import ChatMessageStreamDelta
 from src.insurance_assistants.agents import manager_agent
 from src.insurance_assistants.consts import PRIMARY_HEADING, PROJECT_ROOT_DIR, PROMPT_PREFIX
 
-load_dotenv(override=True)
+# load_dotenv(override=True)
 
 # Setup logger
 logging.basicConfig(level=logging.INFO)
@@ -86,8 +86,10 @@ class UI:
                 intermediate_text += step_log.content or ""
                 yield intermediate_text
 
-    def interact_with_agent(self, prompt, messages, session_state):
+    def interact_with_agent(self, prompt, messages, session_state, api_key):
         # Get or create session-specific agent
+        if not api_key or not api_key.startswith("hf"):
+            raise ValueError("Incorrect API key")
         if "agent" not in session_state:
             # session_state["agent"] = CodeAgent(tools=[], model=InfenceClientModel())
             session_state["agent"] = manager_agent
@@ -220,6 +222,8 @@ class UI:
 """
                         )
                         with gr.Group():
+                            api_key = gr.Textbox(placeholder="Enter your HuggingFace Inference API KEY HERE", 
+                                                 label="API key",show_label=True, type='password')
                             gr.Markdown(
                                 value="**Your question, please...**", container=True
                             )
@@ -295,7 +299,7 @@ class UI:
                     ).then(
                         fn=self.interact_with_agent,
                         # Include session_state in function calls
-                        inputs=[stored_messages, chatbot, session_state],
+                        inputs=[stored_messages, chatbot, session_state, api_key],
                         outputs=[chatbot],
                     ).then(
                         fn=lambda: (
@@ -315,7 +319,7 @@ class UI:
                     ).then(
                         fn=self.interact_with_agent,
                         # Include session_state in function calls
-                        inputs=[stored_messages, chatbot, session_state],
+                        inputs=[stored_messages, chatbot, session_state, api_key],
                         outputs=[chatbot],
                     ).then(
                         fn=lambda: (