Update app.py

app.py

@@ -13,6 +13,8 @@ sys.path.append(os.path.abspath(os.path.dirname(__file__)))
 
 print("Current working directory:", os.getcwd())
 print("Absolute path of app.py:", os.path.abspath(os.path.dirname(__file__)))
+utils_path = os.path.abspath('utils')
+print(f"The absolute path of 'utils' is: {utils_path}")
 
 hf_hub_download(
     repo_id="Comfy-Org/stable-diffusion-v1-5-archive",
@@ -91,9 +93,9 @@ def add_extra_model_paths() -> None:
         print(
             "Could not import load_extra_path_config from main.py. Looking in utils.extra_config instead."
         )
-        # from utils.extra_config import load_extra_path_config
-        sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "utils")))
-        from extra_config import load_extra_path_config
+        from utils.extra_config import load_extra_path_config
+        # sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "utils")))
+        # from extra_config import load_extra_path_config
 
     extra_model_paths = find_path("extra_model_paths.yaml")
 
@@ -183,33 +185,31 @@ def generate_image(positive_prompt, negative_prompt):
             )
             saved_path=f"output/{saveimage_9['ui']['images'][0]['filename']}"
             return saved_path
-
+            
+# Start your Gradio app
+with gr.Blocks() as app:
+    # Add a title
+    gr.Markdown("# FLUX Style Shaping")
+
+    with gr.Row():
+        with gr.Column():
+            # Add an input
+            positive_prompt_input = gr.Textbox(label="Positive Prompt", placeholder="Enter your positive prompt here...")
+            negative_prompt_input = gr.Textbox(label="Negative Prompt", placeholder="Enter your negative prompt here...")
+            # The generate button
+            generate_btn = gr.Button("Generate")
+        
+        with gr.Column():
+            # The output image
+            output_image = gr.Image(label="Generated Image")
+
+        # When clicking the button, it will trigger the `generate_image` function, with the respective inputs
+        # and the output an image
+        generate_btn.click(
+            fn=generate_image,
+            inputs=[positive_prompt_input, negative_prompt_input],
+            outputs=[output_image]
+        )
 
 if __name__ == "__main__":
-    # Start your Gradio app
-    with gr.Blocks() as app:
-        # Add a title
-        gr.Markdown("# FLUX Style Shaping")
-
-        with gr.Row():
-            with gr.Column():
-                # Add an input
-                positive_prompt_input = gr.Textbox(label="Positive Prompt", placeholder="Enter your positive prompt here...")
-                negative_prompt_input = gr.Textbox(label="Negative Prompt", placeholder="Enter your negative prompt here...")
-                # The generate button
-                generate_btn = gr.Button("Generate")
-            
-            with gr.Column():
-                # The output image
-                output_image = gr.Image(label="Generated Image")
-
-            # When clicking the button, it will trigger the `generate_image` function, with the respective inputs
-            # and the output an image
-            generate_btn.click(
-                fn=generate_image,
-                inputs=[positive_prompt_input, negative_prompt_input],
-                outputs=[output_image]
-            )
         app.launch(share=True)
-
-