vhup 0.2__tar.gz → 0.4__tar.gz

{vhup-0.2 → vhup-0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vhup
-Version: 0.2
+Version: 0.4
 Summary: Vho: İşlevsel ve kolaylaştırılmış görsel oluşturma kütüphanesi
 Author: Bedirhan
 Author-email: bedirhan.oytpass@gmail.com
@@ -31,6 +31,9 @@ Requires-Dist: face_recognition
 Requires-Dist: Pillow
 Requires-Dist: insightface>=0.7.3
 Requires-Dist: onnxruntime-gpu
+Requires-Dist: peft
+Requires-Dist: sentencepiece
+Requires-Dist: protobuf
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{vhup-0.2 → vhup-0.4}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as f:
 
 setup(
     name="vhup",
-    version="0.2",
+    version="0.4",
     packages=find_packages(),
     install_requires=[
         "torch>=2.0.0",
@@ -23,7 +23,10 @@ setup(
         "face_recognition",
         "Pillow",
         "insightface>=0.7.3",
-        "onnxruntime-gpu"
+        "onnxruntime-gpu",
+        "peft",
+        "sentencepiece",
+        "protobuf"
         ],
     description="Vho: İşlevsel ve kolaylaştırılmış görsel oluşturma kütüphanesi",
     long_description=README,

vhup-0.4/vhup/__init__.py

@@ -0,0 +1,2 @@
+from .vhup import create_text
+from .horba import create_hore, create_hore_Bef

vhup-0.4/vhup/horba.py

@@ -0,0 +1,110 @@
+import torch
+import random
+import os
+from PIL import Image
+from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, DPMSolverMultistepScheduler
+from deep_translator import GoogleTranslator
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+dtype = torch.float16 if device == "cuda" else torch.float32
+
+FIXED_OUTPUT_FILE = "vho1.png"
+
+translator = GoogleTranslator(source="tr", target="en")
+
+TR_STYLE_MAP = {
+    "gerçekçi": "realistic",
+    "sinematik": "cinematic lighting",
+    "anime": "anime style",
+    "çizgi film": "cartoon style",
+    "karanlık": "dark dramatic mood",
+    "neon": "neon cyberpunk lighting",
+    "portre": "portrait photography",
+    "detaylı": "highly detailed",
+    "fantastik": "fantasy art",
+    "bilim kurgu": "science fiction",
+    "minimal": "minimalist composition"
+}
+
+def load_pipelines():
+    model_id = "runwayml/stable-diffusion-v1-5"
+    pipe = StableDiffusionPipeline.from_pretrained(
+        model_id, 
+        torch_dtype=dtype, 
+        safety_checker=None
+    ).to(device)
+    
+    pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter_sd15.bin")
+    pipe.set_ip_adapter_scale(0.7)
+    
+    pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+    pipe.enable_attention_slicing()
+    return pipe
+
+_pipe = load_pipelines()
+
+def _prepare_prompt(prompt_tr: str) -> str:
+    clean_prompt = prompt_tr.replace("#", "the person")
+    translated = translator.translate(clean_prompt)
+    prompt_lower = prompt_tr.lower()
+    
+    styles = [en for tr, en in TR_STYLE_MAP.items() if tr in prompt_lower]
+    
+    parts = [translated]
+    if styles:
+        parts.append(", ".join(styles))
+    parts.append("masterpiece, ultra high quality, sharp focus, professional lighting")
+    
+    return ", ".join(parts)
+
+def create_hore(prompt: str, hero: str, kalite: float = 7.5) -> str:
+    final_prompt = _prepare_prompt(prompt)
+    face_image = Image.open(hero).convert("RGB")
+    
+    seed = random.randint(0, 999999999)
+    generator = torch.Generator(device=device).manual_seed(seed)
+
+    image = _pipe(
+        prompt=final_prompt,
+        ip_adapter_image=face_image,
+        negative_prompt="worst quality, low quality, blurry, bad anatomy, deformed",
+        num_inference_steps=30,
+        guidance_scale=kalite,
+        generator=generator
+    ).images[0]
+
+    if os.path.exists(FIXED_OUTPUT_FILE):
+        os.remove(FIXED_OUTPUT_FILE)
+        
+    image.save(FIXED_OUTPUT_FILE)
+    return os.path.abspath(FIXED_OUTPUT_FILE)
+
+def create_hore_Bef(prompt: str, hero: str, strength: float = 0.6) -> str:
+    if not os.path.exists(FIXED_OUTPUT_FILE):
+        return create_hore(prompt, hero)
+    
+    prev_image = Image.open(FIXED_OUTPUT_FILE).convert("RGB")
+    face_image = Image.open(hero).convert("RGB")
+    final_prompt = _prepare_prompt(prompt)
+
+    img2img_pipe = StableDiffusionImg2ImgPipeline(**_pipe.components)
+    
+    seed = random.randint(0, 999999999)
+    generator = torch.Generator(device=device).manual_seed(seed)
+
+    image = img2img_pipe(
+        prompt=final_prompt,
+        image=prev_image,
+        ip_adapter_image=face_image,
+        strength=strength,
+        num_inference_steps=30,
+        negative_prompt="worst quality, low quality, blurry",
+        generator=generator
+    ).images[0]
+
+    if os.path.exists(FIXED_OUTPUT_FILE):
+        os.remove(FIXED_OUTPUT_FILE)
+
+    image.save(FIXED_OUTPUT_FILE)
+    return os.path.abspath(FIXED_OUTPUT_FILE)
+    

{vhup-0.2 → vhup-0.4}/vhup.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vhup
-Version: 0.2
+Version: 0.4
 Summary: Vho: İşlevsel ve kolaylaştırılmış görsel oluşturma kütüphanesi
 Author: Bedirhan
 Author-email: bedirhan.oytpass@gmail.com
@@ -31,6 +31,9 @@ Requires-Dist: face_recognition
 Requires-Dist: Pillow
 Requires-Dist: insightface>=0.7.3
 Requires-Dist: onnxruntime-gpu
+Requires-Dist: peft
+Requires-Dist: sentencepiece
+Requires-Dist: protobuf
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{vhup-0.2 → vhup-0.4}/vhup.egg-info/SOURCES.txt

@@ -2,7 +2,8 @@ LICENSE
 README.md
 setup.py
 vhup/__init__.py
-vhup/vhub.py
+vhup/horba.py
+vhup/vhup.py
 vhup.egg-info/PKG-INFO
 vhup.egg-info/SOURCES.txt
 vhup.egg-info/dependency_links.txt

{vhup-0.2 → vhup-0.4}/vhup.egg-info/requires.txt

@@ -14,3 +14,6 @@ face_recognition
 Pillow
 insightface>=0.7.3
 onnxruntime-gpu
+peft
+sentencepiece
+protobuf

vhup-0.2/vhup/__init__.py

@@ -1 +0,0 @@
-from .vhup import create_text