# core/gemini_handler.py
import google.generativeai as genai
import json
import re # For cleaning JSON output

class GeminiHandler:
    def __init__(self, api_key):
        genai.configure(api_key=api_key)
        self.model = genai.GenerativeModel('gemini-1.5-flash-latest') # Or your preferred Gemini model

    def _clean_json_response(self, text_response):
        # Gemini might sometimes wrap JSON in ```json ... ```
        match = re.search(r"```json\s*([\s\S]*?)\s*```", text_response)
        if match:
            return match.group(1).strip()
        return text_response.strip()

    def generate_story_breakdown(self, prompt_text):
        try:
            response = self.model.generate_content(prompt_text)
            cleaned_response = self._clean_json_response(response.text)
            # print(f"Debug: Cleaned Story Breakdown JSON: {cleaned_response}") # For debugging
            story_scenes = json.loads(cleaned_response)
            return story_scenes
        except Exception as e:
            print(f"Error generating story breakdown: {e}")
            print(f"Problematic Gemini Response: {response.text if 'response' in locals() else 'No response object'}")
            return None

    def generate_image_prompt(self, prompt_text):
        try:
            response = self.model.generate_content(prompt_text)
            # Image prompts are usually just text, no need for JSON cleaning unless specified
            return response.text.strip()
        except Exception as e:
            print(f"Error generating image prompt: {e}")
            return None