Spaces:
Sleeping
Sleeping
| from typing import Optional | |
| from google import genai | |
| from google.genai import types | |
| import requests | |
| import os | |
| def analyze_image(image_url: str, analysis_prompt: Optional[str] = None) -> str: | |
| """ | |
| Analyze images using Google Gemini API. Creates summary of the image content. Supports various iamge formats including PNG, JPG, JPEG, etc. | |
| Args: | |
| image_url (str): Url path to the image file to anlyze | |
| analysis_prompt (Optional[str]): Optional prompt for specific analysis focus | |
| Returns: | |
| str: Text containing analysis results | |
| """ | |
| try: | |
| # Initialize Google Gen client | |
| gemini_llm = genai.Client(api_key=os.getenv("GOOGLE_API_KEY")) | |
| print(f"Analyzing image from URL {image_url}") | |
| text=analysis_prompt or "Provide a detailed description of this image." | |
| image_bytes = requests.get(image_url).content | |
| image = types.Part.from_bytes(data=image_bytes, mime_type="image/png") | |
| # Get response from Gemini Flash 2.0 Vision | |
| response = gemini_llm.models.generate_content( | |
| model="gemini-2.0-flash", | |
| contents=[text, image], | |
| ) | |
| print(response.text) | |
| return response.text | |
| except Exception as e: | |
| return {"error": f"Error analyzing image: {str(e)}"} |