Update prompt_expander.py

prompt_expander.py

@@ -19,6 +19,8 @@ SYSTEM_PROMPT = (
 )
 
 class GeminiPromptExpander(ModularPipelineBlocks):
+    model_name = "flux"
+    
     def __init__(self, model_id="gemini-2.5-flash-lite", system_prompt=SYSTEM_PROMPT):
         super().__init__()
         api_key = os.getenv("GOOGLE_API_KEY")
@@ -66,10 +68,10 @@ class GeminiPromptExpander(ModularPipelineBlocks):
         block_state = self.get_block_state(state)
 
         old_prompt = block_state.prompt
-        # print(f"Actual prompt: {old_prompt}")
+        print(f"Actual prompt: {old_prompt}")
         block_state.prompt = self.model.generate_content(old_prompt).text
         block_state.old_prompt = old_prompt
-        # print(f"{block_state.prompt=}")
+        print(f"{block_state.prompt=}")
         self.set_block_state(state, block_state)
 
         return components, state