Update app.py

app.py

@@ -27,11 +27,44 @@ OPENROUTER_MODELS = [
 # ── Sidebar controls ───────────────────────────────────────────────────────────
 with st.sidebar:
     st.subheader("Model & Generation Settings")
+
+    # Model choices
+    OPENROUTER_MODELS = [
+        "openai/gpt-oss-20b:free",
+        "moonshotai/kimi-k2:free",
+        "google/gemini-2.0-flash-exp:free",
+        "google/gemma-3-27b-it:free",
+    ]
     model_choice = st.selectbox("OpenRouter model", OPENROUTER_MODELS, index=0)
     custom_model = st.text_input("Custom model id (optional)", placeholder="e.g. meta-llama/llama-3.1-8b-instruct")
-    temperature = st.slider("Temperature", 0.0, 1.0, 0.1, 0.05)
-    top_p = st.slider("Top-p", 0.0, 1.0, 0.5, 0.05)
 
+    st.markdown("### Preset")
+    PRESETS = {
+        "Extractive (stable)": {"temperature": 0.1, "top_p": 0.9, "desc": "Most deterministic; best for IE"},
+        "Balanced":            {"temperature": 0.2, "top_p": 0.9, "desc": "Slightly more recall"},
+        "Exploratory":         {"temperature": 0.4, "top_p": 0.95, "desc": "More ideas; may add noise"},
+    }
+    preset_names = list(PRESETS.keys())
+    preset = st.selectbox("Choose a preset", preset_names, index=0,
+                          help=PRESETS[preset_names[0]]["desc"])
+
+    # Apply preset button updates the sliders below
+    if st.button("Apply preset"):
+        st.session_state.temperature = PRESETS[preset]["temperature"]
+        st.session_state.top_p = PRESETS[preset]["top_p"]
+        st.toast(f"Applied: {preset}", icon="✅")
+
+    # Sliders are bound to session state so the button can set them
+    temperature = st.slider(
+        "Temperature", 0.0, 1.0, key="temperature", step=0.05,
+        help="Lower = more deterministic; higher = more variety"
+    )
+    top_p = st.slider(
+        "Top-p", 0.0, 1.0, key="top_p", step=0.05,
+        help="Nucleus sampling threshold; 0.9 is a good default"
+    )
+
+    # Ontology controls
     st.markdown("### Ontology (labels)")
     labels_text = st.text_area(
         "Comma-separated labels",
@@ -42,7 +75,6 @@ with st.sidebar:
         "Relationships (comma-separated)",
         value="Relation between any pair of Entities",
     )
-
 # ── Helpers ────────────────────────────────────────────────────────────────────
 def parse_labels(text: str):
     return [lbl.strip() for lbl in text.split(",") if lbl.strip()] or [