Update orchestrator/gemini.py

orchestrator/gemini.py

@@ -1,18 +1,16 @@
 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"):
+def gemini_generate_code(prompt, model="gemini-1.5-pro"):
     api_key = get_gemini_api_key()
     if not api_key:
         raise RuntimeError("Gemini API key not set!")
@@ -21,5 +19,12 @@ def gemini_generate(prompt, model="gemini-1.5-flash"):
     genai.configure(api_key=api_key)
     model_obj = genai.GenerativeModel(model)
     response = model_obj.generate_content(prompt)
-    # Correct line:
     return response.text if hasattr(response, "text") else str(response)
+
+def gemini_explain_code(code, model="gemini-1.5-pro"):
+    # Ask Gemini to explain code in plain English
+    explanation_prompt = (
+        "Explain step-by-step, in clear language, what this Python MCP server code does, and draw a high-level block diagram in Mermaid format:\n\n" +
+        code
+    )
+    return gemini_generate_code(explanation_prompt, model)