Updated

kalam_data.py

@@ -20,10 +20,17 @@ theme_list = theme.keys()
 layer_list = layer.keys()
 surah_list = [theme[key]['surah'] for key in theme]
 
-import gradio as ui; import requests, logging, os; from io import BytesIO
+import gradio as ui; import requests, logging, os, json; from io import BytesIO
 from PIL import Image, ImageEnhance; from requests.exceptions import *; import concurrent.futures
 logging.basicConfig(level=logging.DEBUG)
 
+def expand(expand):
+    pfix  = '(The prompt must intricately describe every part of the image in concrete, objective detail. THINK about what the end goal of the description is, and extrapolate that to what would make satisfying images.)'
+    head = {'Content-Type': 'application/json'}
+    body = json.dumps({"topic": f"{expand} {pfix}", "word_count": 30, "writing_mode": "Simple", "story_genre": "Descriptive"})
+    try: return requests.post(url=os.getenv('gpt4'), headers=head, data=body, timeout=15).json()['story']
+    except Exception as e: print(f"Expansion error: {e}"); return expand
+
 def theme_change(a, i: ui.SelectData):
     print(f"Theme changed -> {a}")
     return ui.Dropdown(value=theme[a]['surah'])
@@ -31,9 +38,9 @@ def theme_change(a, i: ui.SelectData):
 def stella(surah, mode):
     data = {
         'model_version': (None, '1'),
-        'prompt': (None, theme[surah]['prompt']),
+        'prompt': (None, expand(theme[surah]['prompt'])),
         'style_id': (None, '128'),
-        'negative_prompt': (None, 'hands, face, eyes, legs, watermark, text, duplicate, double, bad, ugly, bad quality, naked, nude, sexy'),
+        'negative_prompt': (None, 'hands, jpeg'),
         'aspect_ratio': (None, '1:1'),
         'high_res_results': (None, '1'),
         'cfg': (None, '9.5'),