Spaces:
Sleeping
Sleeping
| """ | |
| project @ LearnableAI | |
| created @ 2025-01-17 | |
| author @ github.com/ishworrsubedii | |
| """ | |
| import google.generativeai as genai | |
| from PIL import Image | |
| class ImageDescriptionGenerator: | |
| def __init__(self): | |
| self.model = genai.GenerativeModel('gemini-1.5-flash') | |
| def process_image(self, image_path, difficulty_level, ): | |
| try: | |
| img = Image.open(image_path) | |
| prompt = self._create_analysis_prompt(difficulty_level) | |
| response = self.model.generate_content([prompt, img]) | |
| return response.text | |
| except Exception as e: | |
| return f"Error processing image: {str(e)}" | |
| def _create_analysis_prompt(self, difficulty_level): | |
| return f""" | |
| # Image Analysis for Children | |
| Please analyze this image and create child-friendly content based on these parameters: | |
| - Difficulty Level: {difficulty_level} | |
| Generate the following: | |
| 1. Simple description of the image (2-3 sentences) | |
| 2. Three interesting details about what you see | |
| 3. Four age-appropriate questions about the image | |
| 4. One interactive activity related to the image | |
| Requirements: | |
| - Use simple, clear language appropriate for {difficulty_level} level | |
| - Questions should encourage observation and critical thinking | |
| - Include both easy and slightly challenging questions | |
| - Activity should be engaging and educational | |
| Format the response exactly like this: | |
| DESCRIPTION: | |
| [Image description] | |
| INTERESTING DETAILS: | |
| 1. [Detail 1] | |
| 2. [Detail 2] | |
| 3. [Detail 3] | |
| QUESTIONS: | |
| Q1: [Question 1] | |
| Q2: [Question 2] | |
| Q3: [Question 3] | |
| Q4: [Question 4] | |
| ACTIVITY: | |
| [Activity description] | |
| """ | |