BlenderCodeGeneration/Main.py at main · wfishell/BlenderCodeGeneration · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
from Iterator import *
from Packages import *
from AnimationGeneration import *

if __name__=='__main__':

    Prompt=("Create a python script for a blender animation of cube moving in a cirle")
    # First Instance
    ParentFolder=MassUpload('SimpleTest')
    ParentFolderID=ParentFolder.create_subfolder()
    SubFolderID_LST=[]
    for i in range (0,3):
        start_time = time.time()
        if i==0:
            AnimationInstance = animationgeneartion(Prompt,
                                                    F"{ParentFolder.FolderName}{i}.blend")
            print(AnimationInstance.prompt)
            code, ExtractedCode = AnimationInstance.chat_with_LLM(AnimationInstance.prompt)
            SubFolder = MassUpload(F"{ParentFolder.FolderName}{i}",ParentFolderID)
            SubFolderID = SubFolder.create_subfolder()
            fileupload = UploadFile(AnimationInstance.filename)
            fileupload.upload_file_to_drive(fileupload.filename, fileupload.filename, "application/x-blender",
                                            SubFolderID)
            fileupload = UploadFile(F"{ParentFolder.FolderName}{i}.mp4")
            fileupload.upload_file_to_drive(fileupload.filename, fileupload.filename, "video/mp4", SubFolderID)
            end_time = time.time()
            elasped_time = end_time - start_time

            # Upload JSON File
            AnimationInstance.json_file['elapsed_time'] = elasped_time
            AnimationInstance.json_file['Code'] = ExtractedCode
            prompt_file = F"{ParentFolder.FolderName}{i}.json"
            with open(prompt_file, "w") as file:
                # Write some text to the file
                json.dump(AnimationInstance.json_file, file, indent=4)
            fileupload = UploadFile(F"{ParentFolder.FolderName}{i}.json")
            fileupload.upload_file_to_drive(fileupload.filename, fileupload.filename, "application/json", SubFolderID)
            SubFolderID_LST.append(SubFolderID)
        elif i==1:
            FirstCycleInstance=FirstCycle(F"{ParentFolder.FolderName}{i-1}.mp4",SubFolderID_LST[0],Prompt)
            ImprovementPlan=FirstCycleInstance.FirstCycle()
            FirstInstanceCode=GetCode(SubFolderID_LST[0])
            AnimationInstance = animationgeneartion(F"Here is the original Prompt {AnimationInstance.prompt}"
                                                                  F" and these are the critques {ImprovementPlan}",
                                                    F"{ParentFolder.FolderName}{i}.blend",
                                                    FirstInstanceCode)
            print(AnimationInstance.prompt)
            code, ExtractedCode = AnimationInstance.chat_with_LLM(AnimationInstance.prompt)
            SubFolder = MassUpload(F"{ParentFolder.FolderName}{i}",ParentFolderID)
            SubFolderID = SubFolder.create_subfolder()
            fileupload = UploadFile(AnimationInstance.filename)
            fileupload.upload_file_to_drive(fileupload.filename, fileupload.filename, "application/x-blender",
                                            SubFolderID)
            fileupload = UploadFile(F"{ParentFolder.FolderName}{i}.mp4")
            fileupload.upload_file_to_drive(fileupload.filename, fileupload.filename, "video/mp4", SubFolderID)
            end_time = time.time()
            elasped_time = end_time - start_time

            # Upload JSON File
            AnimationInstance.json_file['elapsed_time'] = elasped_time
            AnimationInstance.json_file['Code'] = ExtractedCode
            prompt_file = F"{ParentFolder.FolderName}{i}.json"
            with open(prompt_file, "w") as file:
                # Write some text to the file
                json.dump(AnimationInstance.json_file, file, indent=4)
            fileupload = UploadFile(F"{ParentFolder.FolderName}{i}.json")
            fileupload.upload_file_to_drive(fileupload.filename, fileupload.filename, "application/json", SubFolderID)
            SubFolderID_LST.append(SubFolderID)
        else:
            InstanceOfCycle=FullCycle(F"{ParentFolder.FolderName}{i-2}.mp4",F"{ParentFolder.FolderName}{i-1}.mp4",
                      SubFolderID_LST[i-2],SubFolderID_LST[i-1],Prompt)
            PreferredCode,ImprovementPlan=InstanceOfCycle.Cylce()
            AnimationInstance = animationgeneartion(Prompt,
                                                    F"{ParentFolder.FolderName}{i}.blend",
                                                    FirstInstanceCode)
            print(AnimationInstance.prompt)
            code, ExtractedCode = AnimationInstance.chat_with_LLM(F"Here is the original Prompt {AnimationInstance.prompt}"
                                                                  F" and these are the critques {ImprovementPlan}")
            SubFolder = MassUpload(F"{ParentFolder.FolderName}{i}",ParentFolderID)
            SubFolderID = SubFolder.create_subfolder()
            fileupload = UploadFile(AnimationInstance.filename)
            fileupload.upload_file_to_drive(fileupload.filename, fileupload.filename, "application/x-blender",
                                            SubFolderID)
            fileupload = UploadFile(F"{ParentFolder.FolderName}{i}.mp4")
            fileupload.upload_file_to_drive(fileupload.filename, fileupload.filename, "video/mp4", SubFolderID)
            end_time = time.time()
            elasped_time = end_time - start_time

            # Upload JSON File
            AnimationInstance.json_file['elapsed_time'] = elasped_time
            AnimationInstance.json_file['Code'] = ExtractedCode
            prompt_file = F"{ParentFolder.FolderName}{i}.json"
            with open(prompt_file, "w") as file:
                # Write some text to the file
                json.dump(AnimationInstance.json_file, file, indent=4)
            fileupload = UploadFile(F"{ParentFolder.FolderName}{i}.json")
            fileupload.upload_file_to_drive(fileupload.filename, fileupload.filename, "application/json", SubFolderID)
            SubFolderID_LST.append(SubFolderID)