add verbose

prompt_assistant.py

@@ -48,7 +48,7 @@ class Text2ImagePromptAssistant(Tool):
     outputs = ['image']
     description = PROMPT_ASSISTANT_DESCRIPTION
     
-    def __init__(self, *args, openai_api_key='', model_name='text-davinci-003', temperature=0.3, **hub_kwargs):
+    def __init__(self, *args, openai_api_key='', model_name='text-davinci-003', temperature=0.3, verbose=False, **hub_kwargs):
         super().__init__()
         os.environ['OPENAI_API_KEY'] = openai_api_key
         if model_name == 'text-davinci-003':
@@ -64,6 +64,7 @@ class Text2ImagePromptAssistant(Tool):
         set_global_service_context(service_context)
         
         self.storage_path = os.path.dirname(__file__)
+        self.verbose = verbose
         self.hub_kwargs = hub_kwargs
 
     def setup(self):
@@ -84,6 +85,10 @@ class Text2ImagePromptAssistant(Tool):
             self.setup()
 
         better_prompt = str(self.query_engine.query(prompt)).strip()
+        
+        if self.verbose:
+            print('==New prompt generated by LlamaIndex==', flush=True)
+            print(better_prompt, '\n', flush=True)
 
         return self.text2image(better_prompt)