Bug fix

.gitignore

@@ -0,0 +1,2 @@
+images/
+.venv/

README.md

@@ -1,5 +1,5 @@
 ---
-title: CtB AI HP Python 1.0
+title: CtB-AI-castles-hp
 emoji: 🏰
 colorFrom: green
 colorTo: indigo

src/gradio_interface.py

@@ -4,31 +4,35 @@ from src.img_gen import generate_image
 
 
 # Gradio Interface
-def generate_interface(left_hp, right_hp):
-    # Generate the image
-    image = generate_image(left_hp, right_hp)
+# def generate_interface(left_hp, right_hp):
+#     # Generate the image
+#     image = generate_image(left_hp, right_hp)
 
-    if isinstance(image, str):
-        return image  # Return error message
-    else:
-        # Save the image with a timestamped filename
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        output_filename = f"{timestamp}_left_{left_hp}_right_{right_hp}.png"
-        image.save(output_filename)
-        return output_filename
+#     if isinstance(image, str):
+#         return image  # Return error message
+#     else:
+#         # Save the image with a timestamped filename
+#         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+#         output_filename = f"{timestamp}_left_{left_hp}_right_{right_hp}.png"
+#         image.save(output_filename)
+#         return output_filename
 
 # Gradio UI Components
 with gr.Blocks() as demo:
-    gr.Markdown("# CtB - Castle Image Generator")
+    gr.Markdown("#  ========== CtB AI Castle HP ==========")
     with gr.Row():
         left_hp = gr.Slider(0, 100, value=100, label="Left Castle HP")
         right_hp = gr.Slider(0, 100, value=100, label="Right Castle HP")
-    generate_button = gr.Button("Generate Image")
-    output_image = gr.Image(label="Generated Image")
+    with gr.Row():
+        generate_button = gr.Button("Generate Image")
+    with gr.Row():
+        output_image = gr.Image(label="Generated Image")
+    with gr.Row():
+        status_text = gr.Textbox(label="Status", placeholder="Waiting for input...", interactive=False)
 
     # Link the button to the function
     generate_button.click(
-        generate_interface,
+        generate_image,
         inputs=[left_hp, right_hp],
-        outputs=output_image
+        outputs=[output_image, status_text]
     )

src/img_gen.py

@@ -3,12 +3,13 @@ from config.prompts import generate_prompt
 from huggingface_hub import InferenceClient
 from config.models import models
 from config.config import api_token
+from datetime import datetime
 
 # Function to generate images based on the HP values
 def generate_image(left_hp, right_hp):
     # Hardcoded parameters
-    height = 512  # Fixed height
     width = 1024  # Fixed width
+    height = 512  # Fixed height
     num_inference_steps = 20  # Fixed inference steps
     guidance_scale = 2.0  # Fixed guidance scale
     seed = random.randint(0, 1000000)  # Random seed
@@ -44,6 +45,21 @@ def generate_image(left_hp, right_hp):
         print(f"Guidance Scale: {guidance_scale}")
         print(f"Seed: {seed}")
         
-        return image
+        #return image
+    except Exception as e:
+        return None, f"An error occurred: {e}"
+
+    # Save the image with a timestamped filename
+    print("-----SAVING-----", image)
+    path = "images"
+    
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    output_filename = f"{path}/{timestamp}_{seed}_left_{left_hp}_right_{right_hp}.png"
+    try:
+        image.save(output_filename)
     except Exception as e:
-        return f"An error occurred: {e}"
+        return None, f"ERROR: Failed to save image. Details: {e}"
+    print("-----DONE!-----")
+    print("-----CALL THE BANNERS!-----")
+
+    return output_filename, "Image generated successfully! Call the banners!"

temp

@@ -0,0 +1,8 @@
+# colab.ipynb
+import sys
+import os
+
+# Add the root folder to the Python path
+root_path = os.path.abspath("")
+if root_path not in sys.path:
+    sys.path.append(root_path)