Update app.py

app.py

@@ -19,21 +19,25 @@ def generate(outline, characters, settings ):
     
     return hikaye
 
-def cover(prompt):
+def cover(prompts):
     api_key = st.secrets['apikey']
     model = "mann-e/Mann-E_Turbo"
     headers = {"Authorization": f"Bearer {api_key}"}
     api_url = f"https://api-inference.huggingface.co/models/{model}"
-    
-    data = {"inputs": prompt}
-    response = requests.post(api_url, headers=headers, json=data)
-        
-    if 'image' in response.headers.get('content-type', '').lower():
-        image = Image.open(BytesIO(response.content))
-        return image
-    else:
-        st.error("Failed to fetch cover image.")
-        return None
+
+    images = []
+    for prompt in prompts:
+        data = {"inputs": prompt}
+        response = requests.post(api_url, headers=headers, json=data)
+
+        if 'image' in response.headers.get('content-type', '').lower():
+            image = Image.open(BytesIO(response.content))
+            images.append(image)
+        else:
+            st.error(f"Failed to fetch image for prompt: {prompt}")
+            images.append(None)
+
+    return images
 
 def parse_story_response(response):
     print("Debug: Raw response from API:", response)
@@ -58,16 +62,15 @@ outline = st.text_area(label="Story Outline")
 settings = st.text_area(label="Setting")
 
 if st.button(label="Generate"):
-    with st.spinner('Generating story and cover image...'):
-        hikaye = generate(outline, characters, settings )
+    with st.spinner('Generating story and cover images...'):
+        hikaye = generate(outline, characters, settings)
         print("Debug: Story generation response:", hikaye)
-        
+
         if hikaye:
             try:
                 hikaye_json = json.loads(hikaye)
             except json.JSONDecodeError as e:
                 st.error(f"Failed to parse JSON response: {e}")
-                st.error(f"Failed to parse JSON response.{hikaye}")
                 st.stop()
 
             title, p1, p2, p3, p4, prmt = parse_story_response(hikaye_json)
@@ -75,27 +78,26 @@ if st.button(label="Generate"):
             if title and p1 and p2 and p3 and p4:
                 st.markdown(f'### {title}')
 
-                # Display cover image if available
-                image = cover(prmt)
-                if image:
-                    
-                    st.image(image ,caption= prmt)
-                else:
-                    st.error("Failed to generate story cover.")
+                # Prepare prompts for each paragraph
+                prompts = [prmt, f"Image for paragraph 1: {p1}", f"Image for paragraph 2: {p2}", f"Image for paragraph 3: {p3}", f"Image for paragraph 4: {p4}"]
+
+                # Generate and display images
+                images = cover(prompts)
+                for i, image in enumerate(images):
+                    if image:
+                        st.image(image, caption=prompts[i])
+                    else:
+                        st.error(f"Failed to generate image {i+1}.")
 
                 # Display paragraphs
                 st.markdown(f'''
                 {p1}
-
                 {p2}
-
                 {p3}
-
                 {p4}
                 ''')
 
             else:
                 st.error("Failed to generate or parse story.")
-
         else:
             st.error("No story data received.")