Update orchestrator/gemini.py

orchestrator/gemini.py

@@ -1,20 +1,14 @@
 import os
 import streamlit as st
 
-# Try to import the Gemini client
 try:
     import google.generativeai as genai
 except ImportError:
     genai = None
 
 def get_gemini_api_key():
-    """
-    Return Gemini API key from Streamlit secrets or environment variable.
-    """
-    # Try Streamlit Secrets (recommended on Spaces)
     if hasattr(st.secrets, "GEMINI_API_KEY"):
         return st.secrets["GEMINI_API_KEY"]
-    # Fallback to environment variable
     return os.getenv("GEMINI_API_KEY")
 
 def ensure_genai():
@@ -34,13 +28,12 @@ def gemini_generate(prompt, model="gemini-1.5-pro"):
     genai.configure(api_key=api_key)
     model_obj = genai.GenerativeModel(model)
     response = model_obj.generate_content(prompt)
-    # Try to return the 'text' property (for Gemini's Python client), fallback to string
     return response.text if hasattr(response, "text") else str(response)
 
 def gemini_generate_code(system_instruction, user_prompt, model="gemini-1.5-pro"):
     """
-    Use Gemini for code generation with a system prompt and a user prompt.
-    Returns: generated code as string.
+    Use Gemini for code generation (NO explicit system/user roles allowed in public API).
+    Concatenates system_instruction + user_prompt into a single prompt string.
     """
     ensure_genai()
     api_key = get_gemini_api_key()
@@ -48,11 +41,6 @@ def gemini_generate_code(system_instruction, user_prompt, model="gemini-1.5-pro"
         raise RuntimeError("Gemini API key not set in Streamlit secrets or environment variable.")
     genai.configure(api_key=api_key)
     model_obj = genai.GenerativeModel(model)
-    # Compose prompt as role-annotated list for best model behavior
-    prompt = [
-        {"role": "system", "parts": [system_instruction]},
-        {"role": "user", "parts": [user_prompt]},
-    ]
+    prompt = f"{system_instruction.strip()}\n\n{user_prompt.strip()}"
     response = model_obj.generate_content(prompt)
-    # Try to return the 'text' property (for Gemini's Python client), fallback to string
     return response.text if hasattr(response, "text") else str(response)