Update app.py

app.py

@@ -5,6 +5,11 @@ from streamlit_cookies_manager import EncryptedCookieManager
 from openai import OpenAI
 import google.generativeai as genai
 
+# -----------------------------------------------------
+# Move st.set_page_config() to the very top
+# -----------------------------------------------------
+st.set_page_config(page_title="AI Model Comparator", layout="wide")
+
 # -----------------------------------------------------
 # Retrieve API keys from Hugging Face Secrets
 # -----------------------------------------------------
@@ -18,19 +23,12 @@ client = OpenAI(api_key=OPENAI_API_KEY)
 genai.configure(api_key=GEMINI_API_KEY)
 gemini_model = genai.GenerativeModel("gemini-pro")
 
-# -----------------------------------------------------
-# Configure Streamlit Page
-# -----------------------------------------------------
-st.set_page_config(page_title="AI Model Comparator", layout="wide")
-st.title("🤖 AI Model Comparator")
-st.subheader("Compare responses across multiple LLMs.")
-
 # -----------------------------------------------------
 # Secure Cookie Manager for Request Limits
 # -----------------------------------------------------
 cookies = EncryptedCookieManager(
     prefix="ai-model-comparator/",
-    password=os.environ.get("COOKIES_PASSWORD", "MySecretPassword")  # Use a strong secret in deployment
+    password=os.environ.get("COOKIES_PASSWORD", "MySecretPassword")
 )
 
 if not cookies.ready():
@@ -75,8 +73,9 @@ with st.sidebar:
 user_prompt = st.text_area("✍️ Enter your prompt:", "Explain quantum computing in simple terms.")
 
 # -----------------------------------------------------
-# API Request Functions
+# API Request Functions (Using st.cache_data for efficiency)
 # -----------------------------------------------------
+@st.cache_data  # Replaces st.cache for caching API responses
 def get_openai_response(prompt):
     try:
         completion = client.chat.completions.create(
@@ -89,6 +88,7 @@ def get_openai_response(prompt):
     except Exception as e:
         return f"❌ OpenAI error: {e}"
 
+@st.cache_data  # Cache Gemini responses
 def get_gemini_response(prompt):
     try:
         response = gemini_model.generate_content(