kopia lustrzana https://github.com/gd3kr/BlenderGPT
cleanup code
AaryaGadekar
rodzic
4929faf0fa
commit
bdb5394ce4
90
__init__.py
90
__init__.py
|
|
@ -4,6 +4,8 @@ import bpy
|
|||
import bpy.props
|
||||
import re
|
||||
import openai
|
||||
|
||||
from .utilities import *
|
||||
bl_info = {
|
||||
"name": "GPT-4 Blender Assistant",
|
||||
"blender": (2, 82, 0),
|
||||
|
|
@ -51,90 +53,6 @@ if libs_path not in sys.path:
|
|||
|
||||
|
||||
|
||||
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_chat_history = bpy.props.CollectionProperty(type=bpy.types.PropertyGroup)
|
||||
bpy.types.Scene.gpt4_model = bpy.props.EnumProperty(
|
||||
name="GPT Model",
|
||||
description="Select the GPT model to use",
|
||||
items=[
|
||||
("gpt-4", "GPT-4", "Use GPT-4"),
|
||||
("gpt-3.5-turbo", "GPT-3.5 Turbo", "Use GPT-3.5 Turbo"),
|
||||
],
|
||||
default="gpt-4",
|
||||
)
|
||||
bpy.types.Scene.gpt4_chat_input = bpy.props.StringProperty(
|
||||
name="Message",
|
||||
description="Enter your message",
|
||||
default="",
|
||||
)
|
||||
bpy.types.Scene.gpt4_button_pressed = bpy.props.BoolProperty(default=False)
|
||||
bpy.types.PropertyGroup.type = bpy.props.StringProperty()
|
||||
bpy.types.PropertyGroup.content = bpy.props.StringProperty()
|
||||
|
||||
def clear_props():
|
||||
del bpy.types.Scene.gpt4_chat_history
|
||||
del bpy.types.Scene.gpt4_chat_input
|
||||
del bpy.types.Scene.gpt4_button_pressed
|
||||
|
||||
def generate_blender_code(prompt, chat_history, context):
|
||||
messages = [{"role": "system", "content": system_prompt}]
|
||||
for message in chat_history[-10:]:
|
||||
if message.type == "assistant":
|
||||
messages.append({"role": "assistant", "content": "```\n" + message.content + "\n```"})
|
||||
else:
|
||||
messages.append({"role": message.type.lower(), "content": message.content})
|
||||
|
||||
# Add the current user message
|
||||
messages.append({"role": "user", "content": "Can you please write Blender code for me that accomplishes the following task: " + prompt + "? \n. Do not respond with anything that is not Python code. Do not provide explanations"})
|
||||
|
||||
|
||||
response = openai.ChatCompletion.create(
|
||||
model=context.scene.gpt4_model,
|
||||
messages=messages,
|
||||
stream=True,
|
||||
max_tokens=1500,
|
||||
)
|
||||
|
||||
try:
|
||||
collected_events = []
|
||||
completion_text = ''
|
||||
# iterate through the stream of events
|
||||
for event in response:
|
||||
if 'role' in event['choices'][0]['delta']:
|
||||
# skip
|
||||
continue
|
||||
if len(event['choices'][0]['delta']) == 0:
|
||||
# skip
|
||||
continue
|
||||
collected_events.append(event) # save the event response
|
||||
event_text = event['choices'][0]['delta']['content']
|
||||
completion_text += event_text # append the text
|
||||
print(completion_text, flush=True, end='\r')
|
||||
completion_text = re.findall(r'```(.*?)```', completion_text, re.DOTALL)[0]
|
||||
completion_text = re.sub(r'^python', '', completion_text, flags=re.MULTILINE)
|
||||
|
||||
return completion_text
|
||||
except IndexError:
|
||||
return None
|
||||
|
||||
def split_area_to_text_editor(context):
|
||||
area = context.area
|
||||
for region in area.regions:
|
||||
if region.type == 'WINDOW':
|
||||
override = {'area': area, 'region': region}
|
||||
bpy.ops.screen.area_split(override, direction='VERTICAL', factor=0.5)
|
||||
break
|
||||
|
||||
new_area = context.screen.areas[-1]
|
||||
new_area.type = 'TEXT_EDITOR'
|
||||
return new_area
|
||||
|
||||
class GPT4_OT_DeleteMessage(bpy.types.Operator):
|
||||
bl_idname = "gpt4.delete_message"
|
||||
bl_label = "Delete Message"
|
||||
|
|
@ -241,7 +159,7 @@ class GPT4_OT_Execute(bpy.types.Operator):
|
|||
)
|
||||
|
||||
def execute(self, context):
|
||||
openai.api_key = get_api_key(context)
|
||||
openai.api_key = get_api_key(context, __name__)
|
||||
# if null then set to env key
|
||||
if not openai.api_key:
|
||||
openai.api_key = os.getenv("OPENAI_API_KEY")
|
||||
|
|
@ -253,7 +171,7 @@ class GPT4_OT_Execute(bpy.types.Operator):
|
|||
context.scene.gpt4_button_pressed = True
|
||||
bpy.ops.wm.redraw_timer(type='DRAW_WIN_SWAP', iterations=1)
|
||||
|
||||
blender_code = generate_blender_code(context.scene.gpt4_chat_input, context.scene.gpt4_chat_history, context)
|
||||
blender_code = generate_blender_code(context.scene.gpt4_chat_input, context.scene.gpt4_chat_history, context, system_prompt)
|
||||
|
||||
message = context.scene.gpt4_chat_history.add()
|
||||
message.type = 'user'
|
||||
|
|
|
|||
|
|
@ -0,0 +1,90 @@
|
|||
import bpy
|
||||
import openai
|
||||
import re
|
||||
import os
|
||||
import sys
|
||||
|
||||
|
||||
def get_api_key(context, addon_name):
|
||||
preferences = context.preferences
|
||||
addon_prefs = preferences.addons[addon_name].preferences
|
||||
return addon_prefs.api_key
|
||||
|
||||
|
||||
def init_props():
|
||||
bpy.types.Scene.gpt4_chat_history = bpy.props.CollectionProperty(type=bpy.types.PropertyGroup)
|
||||
bpy.types.Scene.gpt4_model = bpy.props.EnumProperty(
|
||||
name="GPT Model",
|
||||
description="Select the GPT model to use",
|
||||
items=[
|
||||
("gpt-4", "GPT-4 (powerful, expensive)", "Use GPT-4"),
|
||||
("gpt-3.5-turbo", "GPT-3.5 Turbo (less powerful, cheaper)", "Use GPT-3.5 Turbo"),
|
||||
],
|
||||
default="gpt-4",
|
||||
)
|
||||
bpy.types.Scene.gpt4_chat_input = bpy.props.StringProperty(
|
||||
name="Message",
|
||||
description="Enter your message",
|
||||
default="",
|
||||
)
|
||||
bpy.types.Scene.gpt4_button_pressed = bpy.props.BoolProperty(default=False)
|
||||
bpy.types.PropertyGroup.type = bpy.props.StringProperty()
|
||||
bpy.types.PropertyGroup.content = bpy.props.StringProperty()
|
||||
|
||||
def clear_props():
|
||||
del bpy.types.Scene.gpt4_chat_history
|
||||
del bpy.types.Scene.gpt4_chat_input
|
||||
del bpy.types.Scene.gpt4_button_pressed
|
||||
|
||||
def generate_blender_code(prompt, chat_history, context, system_prompt):
|
||||
messages = [{"role": "system", "content": system_prompt}]
|
||||
for message in chat_history[-10:]:
|
||||
if message.type == "assistant":
|
||||
messages.append({"role": "assistant", "content": "```\n" + message.content + "\n```"})
|
||||
else:
|
||||
messages.append({"role": message.type.lower(), "content": message.content})
|
||||
|
||||
# Add the current user message
|
||||
messages.append({"role": "user", "content": "Can you please write Blender code for me that accomplishes the following task: " + prompt + "? \n. Do not respond with anything that is not Python code. Do not provide explanations"})
|
||||
|
||||
|
||||
response = openai.ChatCompletion.create(
|
||||
model=context.scene.gpt4_model,
|
||||
messages=messages,
|
||||
stream=True,
|
||||
max_tokens=1500,
|
||||
)
|
||||
|
||||
try:
|
||||
collected_events = []
|
||||
completion_text = ''
|
||||
# iterate through the stream of events
|
||||
for event in response:
|
||||
if 'role' in event['choices'][0]['delta']:
|
||||
# skip
|
||||
continue
|
||||
if len(event['choices'][0]['delta']) == 0:
|
||||
# skip
|
||||
continue
|
||||
collected_events.append(event) # save the event response
|
||||
event_text = event['choices'][0]['delta']['content']
|
||||
completion_text += event_text # append the text
|
||||
print(completion_text, flush=True, end='\r')
|
||||
completion_text = re.findall(r'```(.*?)```', completion_text, re.DOTALL)[0]
|
||||
completion_text = re.sub(r'^python', '', completion_text, flags=re.MULTILINE)
|
||||
|
||||
return completion_text
|
||||
except IndexError:
|
||||
return None
|
||||
|
||||
def split_area_to_text_editor(context):
|
||||
area = context.area
|
||||
for region in area.regions:
|
||||
if region.type == 'WINDOW':
|
||||
override = {'area': area, 'region': region}
|
||||
bpy.ops.screen.area_split(override, direction='VERTICAL', factor=0.5)
|
||||
break
|
||||
|
||||
new_area = context.screen.areas[-1]
|
||||
new_area.type = 'TEXT_EDITOR'
|
||||
return new_area
|
||||
Ładowanie…
Reference in New Issue