Merge branch 'main' of https://github.com/uesleibros/OpenGPT

Author: uesleibros

README.md

@@ -83,8 +83,9 @@ import OpenGPT
 You can install this project as module using pip command.
 
 ```shell
-pip install opengpt4
+pip install opengpt4==0.1.1
 ```
+> It's recommend to use 0.1.1 version for now, the latest version is with bugs. If you want to use the latest version just download source code.
 
 After install, for use it's simple, you can work with something model like this:
 

opengpt/__init__.py

@@ -4,6 +4,7 @@
 import yaml
 import sys
 import inspect
+import os
 
 init()
 
@@ -19,6 +20,8 @@ def Print(context: Text, warn: Optional[bool] = False) -> None:
 class OpenGPT:
 	@classmethod
 	def __init__(self: type, provider: Text, type: Optional[Text] = "completion", options: Optional[Union[Dict[Text, Text], None]] = None) -> None:
+	
+		self.__DIR: Text = os.getcwd()	
 		self.__LoadModels()
 		self.__TYPE: Text = type
 		self.__OPTIONS: Union[Dict[Text, Text], None] = options
@@ -55,7 +58,7 @@ def __InitializeModelClass(self: type) -> None:
 
 	@classmethod
 	def __LoadModels(self: type) -> None:
-		self.__DATA: Dict[Text, Text] = yaml.safe_load(open("opengpt/config.yml").read())
+		self.__DATA: Dict[Text, Text] = yaml.safe_load(open(self.__DIR + "/config.yml").read())
 
 	@classmethod
 	def __Verifications(self: type) -> None:
@@ -96,4 +99,4 @@ def __getattr__(self: type, name: Text) -> None:
 			if callable(attr):
 				return lambda *args, **kwargs: attr(*args, **kwargs)
 			return attr
-		raise AttributeError(f"\"OpenGPT\" object has no attribute \"{name}\"")
+		raise AttributeError(f"\"OpenGPT\" object has no attribute \"{name}\"")

opengpt/models/image/hotpot/model.py

@@ -6,6 +6,7 @@
 import requests
 import fake_useragent
 import logging
+import os
 
 init()
 
@@ -23,6 +24,7 @@ class Model:
 	def __init__(self: type, style: Optional[Text] = "Hotpot Art 9") -> None:
 		self._SETUP_LOGGER()
 		self.__LoadStyles()
+		self.__DIR: Text = os.getcwd()
 		self.__session: requests.Session = requests.Session()
 		self.__UNIQUE_ID: str = UniqueID(16)
 		self.STYLE: Text = style
@@ -71,7 +73,7 @@ def UpdateStyle(self: type, style: Text) -> None:
 
 	@classmethod
 	def __LoadStyles(self: type) -> None:
-		self.__DATA: Dict[Text, Text] = yaml.safe_load(open("opengpt/models/image/hotpot/styles.yml").read())
+		self.__DATA: Dict[Text, Text] = yaml.safe_load(open(self.__DIR + "/styles.yml").read())
 
 	@classmethod
 	def __Fields(self: type, *args: Tuple[int, str], **kwargs: Dict[str, Any]) -> Text:
@@ -111,4 +113,4 @@ def Generate(self: type, prompt: Text, width: Optional[int] = 256, height: Optio
 			"style": self.STYLE,
 			"width": __DATA["width"],
 			"height": __DATA["height"]
-		})
+		})

pyproject.toml

@@ -1,20 +1,19 @@
 [tool.poetry]
 name = "opengpt4"
-version = "0.1.1"
+version = "0.1.6"
 description = ""
 authors = []
 license = "GPL-3.0"
 readme = "README.md"
-packages = [{ include = "opengpt" }]
-exclude = ["**/*.txt"]
+packages = [{ include = "opengpt"}, { include = "libraries"}]
 
 [tool.poetry.dependencies]
 python = "^3.7"
 requests = "2.29.0"
 tls-client = "^0.2"
 fake-useragent = "^1.1.3"
 pydantic = "^1.10.7"
-
+PyYAML = "6.0"
 
 [build-system]
 requires = ["poetry-core"]

requirements.txt

@@ -3,3 +3,4 @@ pydantic==1.10.7
 fake-useragent==1.1.3
 requests==2.28.2
 pycryptodome==3.17
+PyYAML==6.0