Fixed openai api key import

README.md

@@ -0,0 +1,11 @@
+---
+title: CBR Athena
+emoji: 👀
+colorFrom: pink
+colorTo: yellow
+sdk: gradio
+sdk_version: 3.40.1
+app_file: app.py
+pinned: false
+license: apache-2.0
+---

app.py

@@ -3,6 +3,7 @@ from dotenv import load_dotenv
 from cbr_athena.Gradio_Test import Gradio_Test
 
 load_dotenv()
+
 gradio_test = Gradio_Test()
 demo = gradio_test.create_demo()
 

cbr_athena/Gradio_Test.py

@@ -6,7 +6,7 @@ from osbot_utils.utils.Misc import list_set
 from cbr_athena.api.Chat_Predict import Chat_Predict
 from cbr_athena.api.Open_API import Open_API
 
-Open_API().setup()
+
 
 
 TITLE = "# Meet Bobby Tables (head of Application Security). v0.2.0"

cbr_athena/api/Chat_Predict.py

@@ -1,12 +1,15 @@
 import openai
 from osbot_utils.utils.Dev import pprint
 
+from cbr_athena.api.Open_API import Open_API
+
 
 class Chat_Predict:
 
     def __init__(self):
         self.last_message = None
         self.last_response = None
+        Open_API().setup()
 
     def default_prompt(self):
         system_prompt = """
@@ -37,10 +40,6 @@ Project that the viewer might be interested in.
         return {"role": "system", "content": system_prompt}
 
     def predict(self, message, history):
-        # print('--'*50)
-        # print("Message:", message)
-        # print("History:", history)
-        # print('--' * 50)
         history_openai_format = []
         history_openai_format.append(self.default_prompt())
         for human, assistant in history:

cbr_athena/api/Open_API.py

@@ -5,7 +5,7 @@ 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'
+OPEN_AI__API_KEY = 'OPEN_AI__API_KEY'
 
 class Open_API:
 
@@ -15,7 +15,7 @@ class Open_API:
     @cache_on_self
     def api_key(self):
         load_dotenv()
-        return getenv(OPEN_API_KEY)
+        return getenv(OPEN_AI__API_KEY)
 
     def create(self):
         history_openai_format = self.messages()

tests/test_Gradio_Test.py

@@ -0,0 +1,4 @@
+class test_Gradio_Test(TestCase):
+
+    def setUp(self):
+        open_api = Open_API()