Update src/streamlit_app.py

src/streamlit_app.py

@@ -18,6 +18,12 @@ def get_headers(api_key: str) -> Dict[str, str]:
 def pretty(obj: Any) -> str:
     return json.dumps(obj, ensure_ascii=False, indent=2)
 
+
+def safe_get_env(name: str, default: str = "") -> str:
+    v = os.getenv(name)
+    return v if v is not None else default
+
+
 def make_client(base_url: str, api_key: str) -> OpenAI:
     return OpenAI(base_url=base_url, api_key=api_key)
 
@@ -38,22 +44,22 @@ def provider_defaults() -> Dict[str, Dict[str, str]]:
     return {
         "OpenAI": {
             "base_url": "https://api.openai.com/v1",
-            "env_key": os.environ.get("OPENAI_API_KEY"),
+            "env_key": "OPENAI_API_KEY",
             "notes": "Supporte Responses API (web_search, image) + Chat/Embeddings/Models.",
         },
         "Groq": {
             "base_url": "https://api.groq.com/openai/v1",
-            "env_key": os.environ.get("GROQ_API_KEY"),
+            "env_key": "GROQ_API_KEY",
             "notes": "OpenAI-compatible pour Chat/Models/Embeddings (selon offre). Pas de web_search OpenAI.",
         },
         "Ollama": {
             "base_url": "https://ollama.com/v1",
-            "env_key": os.environ.get("OLLAMA_API_KEY"),
+            "env_key": "OLLAMA_API_KEY",
             "notes": "OpenAI-compatible local. Models/Chat ok selon config. Embeddings selon modèles dispo.",
         },
         "Albert (Etalab)": {
             "base_url": "https://albert.api.etalab.gouv.fr/v1",
-            "env_key": os.environ.get("ALBERT_API_KEY"),
+            "env_key": "ALBERT_API_KEY",
             "notes": "OpenAI-compatible (selon endpoints activés). Pas de web_search OpenAI.",
         },
     }
@@ -100,11 +106,11 @@ def pick_default(ids: List[str], preferred: List[str]) -> int:
 # Streamlit UI
 # ----------------------------
 st.set_page_config(page_title="LLM API Playground (pédagogique)", layout="wide")
-st.title("Mini-app de requêtes API sur LLMs")
-st.caption("Choisir un provider")
+st.title("🧪 Mini-app de requêtes API sur LLMs")
+st.caption("Choisir un provider → on charge `/models` → chaque onglet propose un menu déroulant des modèles.")
 
 with st.sidebar:
-    st.header("Configuration")
+    st.header("⚙️ Configuration")
     defaults = provider_defaults()
     provider_name = st.selectbox("Fournisseur", list(defaults.keys()), index=0)
 
@@ -264,10 +270,6 @@ with tabs[1]:
             except Exception as e:
                 st.error(f"Erreur: {e}")
 
-    #with colB:
-        #st.write("À retenir")
-        #st.markdown("- Le modèle sélectionné vient de `/models` quand c’est possible.\n- Sinon, saisis l’ID à la main.")
-
 # ----------------------------
 # 3) Chat completion
 # ----------------------------
@@ -317,10 +319,6 @@ with tabs[2]:
             except Exception as e:
                 st.error(f"Erreur: {e}")
 
-    #with colB:
-    #    st.write("À retenir")
-    #    st.markdown("- Même endpoint, providers différents.\n- Le dropdown aide à éviter les IDs de modèles invalides.")
-
 # ----------------------------
 # 4) JSON extraction
 # ----------------------------
@@ -373,10 +371,6 @@ with tabs[3]:
                 st.error(f"Erreur: {e}")
                 st.info("Certains providers/modèles ne supportent pas `response_format`.")
 
-    #with colB:
-    #    st.write("À retenir")
-    #    st.markdown("- Dropdown = modèle valide (quand `/models` répond).\n- `response_format` peut rester non supporté selon provider.")
-
 # ----------------------------
 # 5) OpenAI web_search (Responses API)
 # ----------------------------
@@ -399,11 +393,7 @@ with tabs[4]:
             prompt = st.text_area(
                 "Prompt",
                 value=(
-                    "Voici quels métadonnées bibliographiques :\n"
-                    "- titre : L'enfant et la rivière\n"
-                    "- auteur : Henri Bosco\n"
-                    "- date : 2015\n\n"
-                    "Fais une recherche dans le catalogue CCFr et trouve le nombre de localisations Sudoc."
+                    "Trouve le ppn de l'autorité personne Albert Camus dans le référentiel français Idref."
                 ),
                 height=180,
             )
@@ -476,9 +466,5 @@ with tabs[5]:
                 except Exception as e:
                     st.error(f"Erreur: {e}")
 
-        #with colB:
-        #    st.write("À retenir")
-        #    st.markdown("- Multimodal OpenAI.\n- Dropdown = modèle valide quand la liste est accessible.")
-
 st.markdown("---")
 st.caption("Astuce : si `/models` ne marche pas chez un provider, la fallback text_input permet quand même de tester.")