added ChatGPT integration

app.py

@@ -15,4 +15,4 @@ gradio_test = Gradio_Test()
 demo = gradio_test.create_demo()
 
 if __name__ == "__main__":
-    demo.launch("admin", "pass1234")
+    demo.launch()

requirements.txt

@@ -1 +1,5 @@
-gradio
+git+https://github.com/owasp-sbot/OSBot-Utils.git
+gradio
+openai
+python-dotenv
+# langchain

test_bot/Gradio_Test.py

@@ -1,14 +1,69 @@
 import gradio as gr
+import openai
+from osbot_utils.utils.Dev import pprint
+
+from test_bot.api.Open_API import Open_API
+
+Open_API().setup()
+
+
 
 class Gradio_Test:
 
+
     def __init__(self):
         pass
 
     def title(self):
-        return "# An title .. (testing deployment) v0.1  "
+        return "# Chat GPT Powered demo"
+
+    def default_prompt(self):
+        system_prompt = """You are an AI-powered CISO (Chief Information Security Officer) bot, designed to provide guidance and answer questions related to cybersecurity and information security. You have extensive knowledge in securing systems, protecting data, implementing best practices, and addressing security concerns. Users will seek your assistance for advice, information, and solutions on a wide range of security topics. Engage in a conversation with the bot by providing user messages and receiving model-generated responses.
+
+                        User: Hi, I have some security concerns and questions. Can you help me?
+
+                        CISO Bot:"""
+        return {"role": "system", "content": system_prompt}
+
+    def predict(self, message, history):
+        history_openai_format = []
+        history_openai_format.append(self.default_prompt())
+        for human, assistant in history:
+            history_openai_format.append({"role": "user", "content": human})
+            history_openai_format.append({"role": "assistant", "content": assistant})
+        history_openai_format.append({"role": "user", "content": message})
+
+        pprint(history_openai_format)
+        response = openai.ChatCompletion.create(
+            model='gpt-3.5-turbo',
+            messages=history_openai_format,
+            temperature=1.0,
+            stream=True
+        )
+
+        partial_message = ""
+        for chunk in response:
+            if len(chunk['choices'][0]['delta']) != 0:
+                partial_message = partial_message + chunk['choices'][0]['delta']['content']
+                yield partial_message
 
     def create_demo(self):
+        # def predict(message, history):
+        #     open_api = Open_API().setup()
+        #     return open_api.create()
+
+        #return gr.ChatInterface(self.predict).queue()
+
+        with gr.Blocks() as demo:
+            gr.Markdown(self.title())
+            textbox_input = gr.Textbox(value='Hello, good morning' , render=False)
+            gr.ChatInterface(self.predict, textbox=textbox_input)
+
+        demo.queue()
+        return demo
+        #return
+
+    def create_demo__2(self):
         title = self.title()
 
         with gr.Blocks() as demo:

test_bot/api/Open_API.py

@@ -0,0 +1,43 @@
+from os import getenv
+
+import openai
+from dotenv import load_dotenv
+from openai import ChatCompletion
+from osbot_utils.decorators.methods.cache_on_self import cache_on_self
+
+OPEN_API_KEY = 'OPEN_API_KEY'
+
+class Open_API:
+
+    def __init__(self):
+        pass
+
+    @cache_on_self
+    def api_key(self):
+        load_dotenv()
+        return getenv(OPEN_API_KEY)
+
+    def create(self):
+        history_openai_format = self.messages()
+        response = ChatCompletion.create(
+            model='gpt-3.5-turbo',
+            messages=history_openai_format,
+            temperature=1.0,
+            stream=True
+        )
+
+        return self.parse_response(response)
+
+    def messages(self):
+        return [{"role": "user", "content": 'Hi'}]
+
+    def parse_response(self, response):
+        partial_message = ""
+        for chunk in response:
+            if len(chunk['choices'][0]['delta']) != 0:
+                partial_message = partial_message + chunk['choices'][0]['delta']['content']
+                yield partial_message
+
+    def setup(self):
+        openai.api_key = self.api_key()
+        return self

tests/api/test_Open_API.py

@@ -0,0 +1,28 @@
+from os import getenv
+from unittest import TestCase
+
+import openai
+from osbot_utils.utils.Dev import pprint
+
+from test_bot.api.Open_API import Open_API, OPEN_API_KEY
+
+
+class test_Open_API(TestCase):
+
+    def setUp(self) -> None:
+        self.open_api = Open_API().setup()
+
+    def test___init__(self):
+        assert type(self.open_api) == Open_API
+
+    def test_api_key(self):
+        api_key = self.open_api.api_key()
+        assert api_key is not None
+        assert api_key == getenv(OPEN_API_KEY)
+
+    def test_create(self):
+        response = self.open_api.create()
+        pprint(list(response))
+
+    def test_setup(self):
+        assert openai.api_key == self.open_api.api_key()