Update core/gemini_handler.py

core/gemini_handler.py

@@ -0,0 +1,37 @@
+# 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