Update app.py

app.py

@@ -37,8 +37,26 @@ if writingPrompt:
     response_content = ""
     for chunk in stream:
         response_content += chunk.choices[0].delta.content
-
+    
     st.write(response_content)
 
+    st.write("Here's some more information to get you started:")
+
+    messages2 = [{ "role": "user", "content": f"Who are the main characters in this text, what are some key locations that should be included, and what are the main story beats of this text: {response_content}"}]
+
+    stream2 = client.chat.completions.create(
+        model="microsoft/Phi-3.5-mini-instruct", 
+    	messages=messages2, 
+    	max_tokens=500,
+    	stream=True
+    )
+
+    #Output response
+    response_content_2 = ""
+    for chunk in stream2:
+        response_content_2 += chunk.choices[0].delta.content
+        
+    st.write(response_content_2)
+    
     #Image generation
     summary = summarizer(response_content, max_length=130, min_length=30, do_sample=False)