Create gemini.py

orchestrator/gemini.py

@@ -0,0 +1,24 @@
+import os
+import streamlit as st
+
+try:
+    import google.generativeai as genai
+except ImportError:
+    genai = None
+
+def get_gemini_api_key():
+    # Try Streamlit Secrets, then env var
+    if hasattr(st.secrets, "GEMINI_API_KEY"):
+        return st.secrets["GEMINI_API_KEY"]
+    return os.getenv("GEMINI_API_KEY")
+
+def gemini_generate(prompt, model="gemini-1.5-flash"):
+    api_key = get_gemini_api_key()
+    if not api_key:
+        raise RuntimeError("Gemini API key not set!")
+    if genai is None:
+        raise ImportError("Please install google-generativeai!")
+    genai.configure(api_key=api_key)
+    model_obj = genai.GenerativeModel(model)
+    response = model_obj.generate_content(prompt)
+    return response.text if hasattr(response, \"text\") else str(response)