Update src/image_gen.py

src/image_gen.py

@@ -16,7 +16,7 @@ genai.configure(api_key=GOOGLE_API_KEY)
 client = InferenceClient(provider="hf-inference", api_key=HF_TOKEN)
 
 def split_text_for_images(number_of_images):
-    with open("./data/text/text.txt", "r", encoding="utf-8") as file:
+    with open("./text/text.txt", "r", encoding="utf-8") as file:
         text = file.read().strip()
 
     total_length = len(text)
@@ -101,7 +101,7 @@ def image_gen(number_of_images = 3,detail_level = "short", perspective="neutral"
     texts = split_text_for_images(number_of_images)
     index = 0
     for text in tqdm(texts, desc="Processing", unit="image"):
-        output_path = f"./data/image/{index}.png"
+        output_path = f"./image/{index}.png"
         prompt = describe_image(text, detail_level, perspective, emotion, time_setting, art_style)
         print(prompt)
 
@@ -121,6 +121,6 @@ def image_gen(number_of_images = 3,detail_level = "short", perspective="neutral"
                 print(f"Thử lại sau {wait_time:.2f} giây...")
                 time.sleep(wait_time)
         index += 1
-    os.remove("./data/text/text.txt")
+    os.remove("./text/text.txt")
 if __name__ == "__main__":
     image_gen(number_of_images = 3, detail_level="short", perspective="neutral", emotion="sad", time_setting="classic", art_style="realistic", style="anime", color_palette="monochrome")