add: use case three

app.py

@@ -4,6 +4,7 @@ import tempfile
 import streamlit as st
 
 from src.pipeline.main import LearnableAI
+from src.services.use_case_three.story_generator import StoryGenerator
 
 
 def main():
@@ -14,7 +15,7 @@ def main():
     st.sidebar.header("Configuration")
     use_case = st.sidebar.radio(
         "Select Use Case",
-        ["Use Case: 1(Word-Sentences)", "Use Case: 2(Image Describer)"]
+        ["Use Case: 1(Word-Sentences)", "Use Case: 2(Image Describer)", "Use Case: 3(Story Generator From Image)"]
     )
 
     difficulty = st.sidebar.selectbox(
@@ -53,7 +54,7 @@ def main():
             else:
                 st.warning("Please fill in both words and interest fields!")
 
-    else:  # Image Description
+    elif use_case == "Use Case: 2(Image Describer)":
         st.header("Image Description")
 
         upload_method = st.radio(
@@ -93,5 +94,47 @@ def main():
                     st.error(f"An error occurred while processing the image: {str(e)}")
 
 
+    elif use_case == "Use Case: 3(Story Generator From Image)":
+        st.title("Story Generator from Images")
+        st.subheader("Upload up to 3 images and generate a children's story")
+
+        uploaded_files = st.file_uploader(
+            "Upload up to 3 images",
+            type=["jpg", "jpeg", "png"],
+            accept_multiple_files=True
+        )
+
+        if uploaded_files:
+            if len(uploaded_files) > 3:
+                st.error("You can upload a maximum of 3 images.")
+            else:
+                st.write("Uploaded Images:")
+                for uploaded_file in uploaded_files:
+                    st.image(uploaded_file, caption=uploaded_file.name, use_column_width=True)
+
+                if st.button("Generate Story"):
+                    story_generator = StoryGenerator()
+                    story_parts = []
+
+                    try:
+                        for uploaded_file in uploaded_files:
+                            with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as temp_file:
+                                temp_file.write(uploaded_file.read())
+                                temp_image_path = temp_file.name
+
+                            story_part = story_generator.process_image(temp_image_path)
+                            story_parts.append(story_part)
+
+                            os.remove(temp_image_path)
+
+                        full_story = "\n\n".join(story_parts)
+                        st.success("Story Generated Successfully!")
+                        st.write("### Your Story")
+                        st.write(full_story)
+
+                    except Exception as e:
+                        st.error(f"An error occurred: {str(e)}")
+
+
 if __name__ == "__main__":
     main()

requirements.txt

@@ -1,3 +1,4 @@
 groq
 streamlit
-google-generativeai
+google-generativeai
+replicate

src/pipeline/main.py

@@ -14,8 +14,8 @@ from src.services.use_case_two.image_describer import ImageDescriptionGenerator
 
 class LearnableAI:
     def __init__(self):
-        self.groq_client = Groq(api_key=os.getenv('GROQ_API_KEY'))
-        self.genai = genai.configure(api_key=os.getenv('GENAI_API_KEY'))
+        self.groq_client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+        self.genai = genai.configure(api_key=os.getenv("GENAI_API_KEY"))
 
         self.use_case_one = UseCaseOne(client=self.groq_client)
 

src/services/use_case_three/story_generator.py

@@ -0,0 +1,33 @@
+import google.generativeai as genai
+from PIL import Image
+
+
+class StoryGenerator:
+    def __init__(self):
+        self.model = genai.GenerativeModel('gemini-1.5-flash')
+
+    def process_image(self, image_path):
+        try:
+            img = Image.open(image_path)
+            prompt = self._create_story_prompt()
+
+            response = self.model.generate_content([prompt, img])
+            return response.text
+
+        except Exception as e:
+            return f"Error processing image: {str(e)}"
+
+    def _create_story_prompt(self):
+        return """
+        Create a delightful and imaginative children's story inspired by the following captions from up to 4 images. Combine all captions into a single storyline that flows naturally. Use simple, child-friendly language and a playful tone suitable for children aged 4–8.
+        
+        Ensure the story has:
+        
+        A central character (e.g., a curious child, an animal, or a magical creature) that ties the events together.
+        A clear beginning, middle, and end with an engaging plot.
+        Elements of wonder and adventure to capture a child's imagination.
+        A gentle life lesson or moral, encouraging themes like kindness, courage, or curiosity.
+        Include vivid descriptions of settings and actions to match the imagery, and ensure the captions inspire key moments in the story.
+        
+        Please incorporate all these captions seamlessly into the storyline
+        """