img

app.py

@@ -71,7 +71,7 @@ def create_interface():
             generate_button = gr.Button("Generate Prompt")
         with gr.Row():
             text_output = gr.Textbox(label="LLM Generated Text", lines=10, show_copy_button=True)
-            image_output = gr.Image(label="Generated Image", type="pil")
+            image_output = gr.Image(label="Generated Image", type="filepath")
 
         # Updated Models based on provider
         def update_model_choices(provider):
@@ -168,6 +168,7 @@ def create_interface():
             try:
                 seed = random.randint(0, 1000000)
                 image_path = llm_node.generate_image(text, seed=seed)
+                print(f"Image generated: {image_path}")
                 return image_path
             except Exception as e:
                 print(f"An error occurred while generating the image: {e}")

llm_inference.py

@@ -253,7 +253,8 @@ Your output is only the caption itself, no comments or extra formatting. The cap
                 height=height,
                 api_name="/generate_image"
             )
-            print(result)
-            return result
+            # Extract the image path from the result tuple
+            image_path = result[0]
+            return image_path
         except Exception as e:
             raise Exception(f"Error generating image: {str(e)}")