Feat: Initial commit to prod

README.md

@@ -1,12 +1,14 @@
----
-title: GhibhiGenerator
-emoji: 👁
-colorFrom: indigo
-colorTo: blue
-sdk: gradio
-sdk_version: 5.23.3
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: GhibhiGenerator
+emoji: 🎨
+colorFrom: yellow
+colorTo: purple
+sdk: gradio
+sdk_version: 4.38.1
+app_file: app.py
+pinned: false
+license: agpl-3.0
+short_description: Turn your images into Ghibli style!
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,86 @@
+import random, gradio as gr
+from atelier_generator import AtelierGenerator
+client = AtelierGenerator(gradio=True, wm_text="GhibhiGenerator! by NithiyaShri")
+
+gr_css = """
+footer {
+    display: none !important;
+}
+
+.image-frame.svelte-rrgd5g img {
+    height: 640px;
+    object-fit: contain;
+}
+
+.grid-wrap.svelte-eynlr2.svelte-eynlr2 {
+    overflow-y: auto;
+}
+"""
+
+gr_thm = gr.themes.Default(
+            primary_hue=gr.themes.colors.rose,
+            secondary_hue=gr.themes.colors.rose,
+            neutral_hue=gr.themes.colors.zinc
+        )
+
+lora_list = {
+    'NithiyaShri/studio-ghibli': 'studio-ghibli',
+    'NithiyaShri/anime-plus': 'anime-plus',
+    'NithiyaShri/softserve-anime': 'softserve-anime'
+}
+
+# Initialize a cache dictionary outside the function
+caption_cache = {}
+
+def GhibhiGenerator(src, lora, mem, progress = gr.Progress()):
+    progress(0.1, "Preparing image (stage 1 of 3)...")
+    size, _, _ = client.size_checker(src)
+
+    # Check if the caption is already cached
+    if src in caption_cache:      
+        prompt = caption_cache[src]
+    
+    else:
+        progress(0.5, f"Processing {size} image (stage 2 of 3)...")
+        prompt = client.image_caption(src)
+        caption_cache[src] = prompt
+
+    progress(0.8, f"Rendering {size} image (stage 3 of 3)...")
+    result = client.image_variation(src, prompt, image_size=size, strength=0.33, lora_flux=lora_list.get(lora))
+    
+    if result:
+        mem.insert(0, result)
+        return mem
+    else:
+        gr.Warning('Something went wrong! Please try again.')
+        return mem
+
+with gr.Blocks(analytics_enabled=False, title='GhibhiGenerator by NithiyaShri', css=gr_css, theme=gr_thm) as demo:
+    gr.Markdown("## <br><center>GhibhiGenerator - Transform your images to Ghibli style!")
+    gr.Markdown("<center>Copyright (C) 2025 Nithiya Shri. All rights reserved")
+    
+    with gr.Row():
+        with gr.Column():
+            with gr.Column(variant='panel'):
+                gr.Markdown("## <center>📸 Upload Your Image!")
+                src = gr.Image(type='filepath', label='Source Image', height=640, sources=['upload'])
+            
+            with gr.Column(variant='panel'):
+                lst = gr.Radio(choices=lora_list.keys(), value=list(lora_list.keys())[0], label='Select a style:')
+                run = gr.Button("GhibhiGenerator!", variant='stop')
+
+        with gr.Column():    
+            with gr.Column(variant='panel'):
+                gr.Markdown("## <center>🎨 GhibhiGenerator It!")
+                res = gr.Gallery(label='Result Image', height=814.19)
+                mem = gr.State([])
+   
+    gr.Markdown("<br><center>Disclaimer: Due to huge demand, image processing might take longer than expected.<br>")
+
+    run.click(
+        inputs=[src, lst, mem],
+        outputs=res,
+        fn=GhibhiGenerator
+    )
+
+demo.launch(inbrowser=True)

requirements.txt

@@ -0,0 +1,4 @@
+gradio==4.38.1
+fastapi==0.112.4
+pydantic==2.10.6
+atelier-generator