Updated

kalam_data.py

@@ -24,23 +24,16 @@ 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'])
 
-def stella(expanded, surah, mode):
+def stella(surah, mode):
     data = {
         'model_version': (None, '1'),
-        'prompt': (None, expanded),
+        'prompt': (None, theme[surah]['prompt']),
         'style_id': (None, '128'),
-        'negative_prompt': (None, 'hands, jpeg, bad quality, cartoon, 3d'),
+        'negative_prompt': (None, 'hands, face, eyes, legs'),
         'aspect_ratio': (None, '1:1'),
         'high_res_results': (None, '1'),
         'cfg': (None, '9.5'),
@@ -75,12 +68,11 @@ def stella(expanded, surah, mode):
 def queue(a, b):
     quantities = 4
     result_list = [None] * quantities
-    gpt = expand(theme[a]['prompt'])
     
     with concurrent.futures.ThreadPoolExecutor() as executor:
         futures = []
         for i in range(quantities):
-            future = executor.submit(lambda x: stella(gpt, a, b), i)
+            future = executor.submit(lambda x: stella(a, b), i)
             futures.append(future)
 
     for i, future in enumerate(futures):