Update app.py

app.py

@@ -29,15 +29,16 @@ def txt2story(img_text):
     )
 
     print("Constructing prompt for story generation...")
+    
     content_prompt = f'''Based on the image description "{img_text}", conclude the story. 
     Resolve the conflict or summarize the outcome of the situation. Ensure the story MUST
     have a definitive ending. The end.'''
 
     print("Preparing message sequences for interaction...")
     messages = [
-        {"role": "system", "content": "You are an expert short story writer who specialises in ending stories positively."},
-        {"role": "user", "content": img_text, "temperature": 1},
-        {"role": "system", "content": content_prompt, "temperature": 0.7},
+        {"role": "system", "content": '''You are an expert short story writer who specialises 
+        in writing stories on input set of words.''', "temperature": 0.8},
+        {"role": "user", "content": "f'Here is input set of words: '{img_text}''", "temperature": 1},
     ]
     
     print("Generating story completion using the AI model...")