modified the code, WIP

app.py

@@ -234,12 +234,7 @@ with gr.Blocks() as demo:
 
     regenerate_comps_btn.click(
         core.regenerate_composition_data,
-        inputs=[image_description,
-            narration,
-            character,
-            dialouge,
-            location,
-            setting,
+        inputs=[
             current_episode,
             current_scene,
             current_frame,

core.py

@@ -215,48 +215,54 @@ def toggle_developer_options(is_developer: bool):
 
 
 def regenerate_composition_data(
-    description: str,
-    narration: str,
-    character: str,
-    dialouge: str,
-    location: str,
-    frame_setting: str,
     current_episode: int,
     current_scene: int,
     current_frame: int,
     episodes_data: dict,
 ):
-    pass
-#     print(
-#         f"Generating compositions for episode: {current_episode} and scene: {current_scene} and frame: {current_frame}."
-#     )
-#     frame = episodes_data[current_episode][current_frame]
-#     print(frame)
-#     prompt_dict = {
-#         "system": script_gen.generate_image_compositions_instruction,
-#         "user": jinja2.Template(
-#             source=script_gen.generate_image_compositions_user_prompt
-#         ).render(
-#             {
-#                 "FRAME": frame,
-#             }
-#         ),
-#     }
-
-#     compositions = llm.generate_valid_json_response(prompt_dict)
-#     print(compositions)
-    # frame.compositions = [
-    #     Composition(**composition)
-    #     for composition in compositions["compositions"]
-    # ]
+    print(
+        f"Generating compositions for episode: {current_episode} and scene: {current_scene} and frame: {current_frame}."
+    )
+    
+    # Retrieve the current frame data
+    frame = episodes_data[current_episode][current_frame]
+    print(frame)
+    
+    # Generate the prompt for compositions
+    prompt_dict = {
+        "system": script_gen.generate_image_compositions_instruction,
+        "user": jinja2.Template(
+            source=script_gen.generate_image_compositions_user_prompt
+        ).render(
+            {
+                "FRAME": frame,
+            }
+        ),
+    }
+
+    # Generate compositions using LLM
+    compositions = llm.generate_valid_json_response(prompt_dict)
+    print(compositions)
+
+    # Update frame with new compositions
+    frame.compositions = [
+        Composition(
+            **composition,
+            seed="",  # Set default empty seed
+            image=""  # Set default empty image
+        )  # Create a Composition dataclass from the response
+        for composition in compositions["compositions"]
+    ]
+    
+    # Update the episodes_data dictionary with the modified frame
+    episodes_data[current_episode][current_frame] = frame
+    print(f"Updated frame {current_frame} in episode {current_episode} with new compositions.")
 
 
 def regenerate_data(
     frame_data: ComicFrame,
 ):
     pass
-
-
 #     # for
 #     payload = {
 #         "prompt": composition.prompt,

openai_wrapper.py

@@ -70,7 +70,7 @@ class OpenAIModel(abc.ABC):
         temperature: int = 0.6,
     ) -> str:
         """Generate a response with retries, returning a valid JSON."""
-        for _ in range(parameters.MAX_TRIES):
+        for _ in range(int(parameters.MAX_TRIES)):
             try:
                 model_response = self.generate_response(
                     prompt_dict, max_output_tokens, temperature, {"type": "json_object"}