Update app.py

app.py

@@ -19,7 +19,11 @@ warnings.filterwarnings('ignore')
 def get_env_or_secret(key_name: str, default: str = None):
     """Helper to read from Streamlit secrets first, then env vars."""
     try:
-        return st.secrets.get(key_name, os.getenv(key_name, default))
+        # Try Streamlit secrets first
+        if hasattr(st, 'secrets') and key_name in st.secrets:
+            return st.secrets[key_name]
+        # Fall back to environment variables
+        return os.getenv(key_name, default)
     except Exception:
         return os.getenv(key_name, default)
 
@@ -42,25 +46,17 @@ def get_active_llm_provider():
 def get_llm_summary(prompt: str, context: str = "") -> str:
     """
     Robust LLM summary using AIML /responses endpoint.
-    Changes:
-    - Wraps the prompt inside `input: {"text": ...}` as some endpoints expect.
-    - Ensures `response_format` is set to force textual output.
-    - Keeps the previous temperature-fallback (remove if unsupported).
-    - Retries once with a larger token budget (without temperature) if no text returned.
     """
     # Build final prompt safely
     if context:
-        full_prompt = f"{context}
-
-{prompt}"
+        full_prompt = f"{context}\nUser Query: {prompt}"
     else:
         full_prompt = prompt
 
-    api_key = get_env_or_secret("AI_ML_API_KEY") or os.getenv("AI_ML_API_KEY")
+    api_key = get_env_or_secret("AI_ML_API_KEY")
     if not api_key:
         return (
-            "AI Analysis unavailable — AI_ML_API_KEY not configured.
-"
+            "AI Analysis unavailable — AI_ML_API_KEY not configured.\n"
             "Go to Settings → Secrets → Create secret 'AI_ML_API_KEY' with your AI/ML API key."
         )
 
@@ -78,13 +74,12 @@ def get_llm_summary(prompt: str, context: str = "") -> str:
         except Exception as e:
             return None, f"AI Analysis Request Failed (parsing): {e}"
 
-    # initial payload (wrap input in dict and force text response_format)
+    # Fixed payload with correct model name and input format
     base_payload = {
-        "model": "openai/gpt-5-2025-08-07",
-        "input": {"text": full_prompt},
+        "model": "gpt-4o",  # Changed from "openai/gpt-4" to "gpt-4o"
+        "input": full_prompt,  # Changed from {"text": full_prompt} to just full_prompt
         "max_output_tokens": 1024,
         "temperature": 0.0,
-        "reasoning_effort": "low",
         "response_format": {"type": "text"}
     }
 
@@ -145,19 +140,16 @@ def get_llm_summary(prompt: str, context: str = "") -> str:
                             if isinstance(c, dict) and c.get("text"):
                                 texts.append(c["text"].strip())
 
-        return "
-
-".join(texts).strip()
+        return "\n".join(texts).strip()
 
     out_text = extract_text_from_response(data)
     # If we received only reasoning without text, try an explicit retry that forces text
     if not out_text:
-        # Second attempt: ensure response_format present and input wrapped (no temperature)
+        # Second attempt: try with gpt-4o-mini as fallback
         retry_payload = {
-            "model": base_payload["model"],
-            "input": {"text": full_prompt},
+            "model": "gpt-4o-mini",  # Use mini version as fallback
+            "input": full_prompt,
             "max_output_tokens": 2048,
-            "reasoning_effort": "low",
             "response_format": {"type": "text"}
         }
         data2, err2 = call_api(retry_payload)
@@ -731,4 +723,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()