added a field in addons tab of preferences to input the OpenAI API key

__init__.py

@@ -2,16 +2,6 @@
 import os
 import bpy
 
-# Add the 'libs' folder to the Python path
-libs_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "lib")
-if libs_path not in sys.path:
- sys.path.append(libs_path)
-
-print (libs_path)
-import openai
-
-openai.api_key = os.getenv("OPENAI_API_KEY")
-
 bl_info = {
  "name": "GPT-4 Blender Assistant",
  "blender": (2, 82, 0),
@@ -26,6 +16,20 @@
 }
 
 
+# Add the 'libs' folder to the Python path
+libs_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "lib")
+if libs_path not in sys.path:
+ sys.path.append(libs_path)
+
+print (libs_path)
+import openai
+
+def get_api_key(context):
+ preferences = context.preferences
+ addon_prefs = preferences.addons[__name__].preferences
+ return addon_prefs.api_key
+
+
 def init_props():
  bpy.types.Scene.gpt4_natural_language_input = bpy.props.StringProperty(
  name="Command",
@@ -43,6 +47,8 @@ def clear_props():
 
 def generate_blender_code(prompt):
 
+
+
  try:
  response = openai.ChatCompletion.create(
  model="gpt-4",
@@ -74,11 +80,9 @@ def generate_blender_code(prompt):
  completion_text = ''
  # iterate through the stream of events
  for event in response:
- # check if role key is present in event['choices'][0]['delta']
  if 'role' in event['choices'][0]['delta']:
  # skip
  continue
- # check if length of event['choices'][0]['delta']['content'] is 0
  if len(event['choices'][0]['delta']) == 0:
  # skip
  continue
@@ -130,6 +134,12 @@ class GPT4_OT_Execute(bpy.types.Operator):
  )
 
  def execute(self, context):
+ openai.api_key = get_api_key(context)
+
+ if not openai.api_key:
+ self.report({'ERROR'}, "No API key detected. Please set the API key in the addon preferences.")
+ return {'CANCELLED'}
+
  context.scene.gpt4_button_pressed = True
  bpy.ops.wm.redraw_timer(type='DRAW_WIN_SWAP', iterations=1)
 
@@ -156,15 +166,30 @@ def execute(self, context):
 def menu_func(self, context):
  self.layout.operator(GPT4_OT_Execute.bl_idname)
 
+class GPT4AddonPreferences(bpy.types.AddonPreferences):
+ bl_idname = __name__
+
+ api_key: bpy.props.StringProperty(
+ name="API Key",
+ description="Enter your OpenAI API Key",
+ default="",
+ subtype="PASSWORD",
+ )
+
+ def draw(self, context):
+ layout = self.layout
+ layout.prop(self, "api_key")
 
 def register():
+ bpy.utils.register_class(GPT4AddonPreferences)
  bpy.utils.register_class(GPT4_OT_Execute)
  bpy.utils.register_class(GPT4_PT_Panel)
  bpy.types.VIEW3D_MT_mesh_add.append(menu_func)
  init_props()
 
 
 def unregister():
+ bpy.utils.unregister_class(GPT4AddonPreferences)
  bpy.utils.unregister_class(GPT4_OT_Execute)
  bpy.utils.unregister_class(GPT4_PT_Panel)
  bpy.types.VIEW3D_MT_mesh_add.remove(menu_func)