Create src/config.py

src/config.py

@@ -0,0 +1,28 @@
+import os
+import mimetypes
+
+def read_system_prompt(file_path: str) -> str:
+    with open(file_path, 'r', encoding='utf-8') as file:
+        return file.read()
+
+def load_prompt_template(base_image_path: str, user_instruction: str) -> str:
+    base_prompt = read_system_prompt('prompt.txt')
+    enriched_prompt = base_prompt.format(
+        base_image_path=base_image_path,
+        user_instruction=user_instruction
+    )
+    return enriched_prompt.strip()
+
+def save_image(file_name: str, data: bytes):
+    with open(file_name, "wb") as f:
+        f.write(data)
+    print(f"Fichier sauvegardé : {file_name}")
+
+def save_image(user_text, file_name, data):
+    sanitized_user_text = user_text[:10].replace(" ", "_").lower()
+    new_file_name = f"genere_pour_{sanitized_user_text}_{file_name}"
+    file_path = os.path.join("generated_image", new_file_name)
+    os.makedirs("generated_image", exist_ok=True)
+    with open(file_path, "wb") as f:
+        f.write(data)
+    print(f"Image générée : {file_path}")