Updated

kalam_data.py

@@ -27,8 +27,8 @@ 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({"sentence": f"{expand} {pfix}", "target_word_count": 40})
-    try: return requests.post(url=os.getenv('gpt4'), headers=head, data=body, timeout=15).json()['expanded_sentence']
+    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):