Update main.py

main.py

@@ -11,11 +11,11 @@ class UserInterface():
         st.text("An Open Source Chat Application")
         st.header("Open LLMs")
 
-        # self.API_KEY = st.sidebar.text_input(
-        #     'API Key',
-        #     type='password',
-        #     help="Type in your HuggingFace API key to use this app"
-        # )
+        self.API_KEY = st.sidebar.text_input(
+            'API Key',
+            type='password',
+            help="Type in your HuggingFace API key to use this app"
+        )
 
         models_name = (
             "HuggingFaceH4/zephyr-7b-beta",
@@ -51,19 +51,19 @@ class UserInterface():
             "max_length": self.max_token_length
         }
 
-        # os.environ['HUGGINGFACEHUB_API_TOKEN'] = self.API_KEY/
+        os.environ['HUGGINGFACEHUB_API_TOKEN'] = self.API_KEY
 
     
     def form_data(self):
 
         try:
-            # if not self.API_KEY.startswith('hf_'):
-            #     st.warning('Please enter your API key!', icon='⚠')
-            #     text_input_visibility = True
-            # else:
-            #     text_input_visibility = False
+            if not self.API_KEY.startswith('hf_'):
+                st.warning('Please enter your API key!', icon='⚠')
+                text_input_visibility = True
+            else:
+                text_input_visibility = False
             
-            text_input_visibility = False
+
             if "messages" not in st.session_state:
                     st.session_state.messages = []