Merge pull request #141 from yjg30737/feature/general

feature/general

Author: yjg30737

pyqt_openai/replicate_widget/replicateRightSideBar.py

@@ -36,7 +36,7 @@ def __initVal(self):
         self.__number_of_images_to_create = CONFIG_MANAGER.get_replicate_property('number_of_images_to_create')
         self.__save_prompt_as_text = CONFIG_MANAGER.get_replicate_property('save_prompt_as_text')
 
-        self.__wrapper = ReplicateWrapper(self.__api_key)
+        self.__wrapper = ReplicateWrapper(self.__api_key.strip() if self.__api_key else '')
 
     def __initUi(self):
         self.__apiKeyLineEdit = QLineEdit()
@@ -165,8 +165,10 @@ def __initUi(self):
     def __replicateChanged(self, v):
         sender = self.sender()
         if sender == self.__apiKeyLineEdit:
+            v = v.strip() if v else ''
             self.__api_key = v
             CONFIG_MANAGER.set_replicate_property('REPLICATE_API_TOKEN', v)
+            self.__wrapper.set_api(v)
         elif sender == self.__widthSpinBox:
             self.__width = v
             CONFIG_MANAGER.set_replicate_property('width', v)

pyqt_openai/util/replicate_script.py

@@ -18,6 +18,7 @@ def set_api(self, api_key):
         :param api_key:
         :return:
         """
+        print("API KEY", api_key)
         self.__api_key = api_key
         os.environ['REPLICATE_API_TOKEN'] = self.__api_key
         return True