Update app.py

app.py

@@ -1,15 +1,8 @@
-import os
 import streamlit as st
 import requests
-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
 # -----------------------------------------------------
@@ -19,41 +12,32 @@ GEMINI_API_KEY = st.secrets["GEMINI_API_KEY"]
 # -----------------------------------------------------
 # Initialize OpenAI & Gemini
 # -----------------------------------------------------
-client = OpenAI(api_key=OPENAI_API_KEY)
+client = OpenAI(api_key=OPENAI_API_KEY)  # OpenAI client
 genai.configure(api_key=GEMINI_API_KEY)
 gemini_model = genai.GenerativeModel("gemini-pro")
 
 # -----------------------------------------------------
-# Secure Cookie Manager for Request Limits
+# Configure Streamlit Page
 # -----------------------------------------------------
-cookies = EncryptedCookieManager(
-    prefix="ai-model-comparator/",
-    password=os.environ.get("COOKIES_PASSWORD", "MySecretPassword")
-)
-
-if not cookies.ready():
-    st.stop()  # Wait until cookies are available
-
-# Initialize request count if not set
-if "request_count" not in cookies:
-    cookies["request_count"] = 0
-
-request_count = int(cookies["request_count"])
+st.set_page_config(page_title="AI Model Comparator", layout="wide")
+st.title("🤖 AI Model Comparator")
+st.subheader("Compare responses across multiple LLMs.")
 
 # -----------------------------------------------------
-# Check Request Limit (Max 3 Requests Per User)
+# User Input: Prompt
 # -----------------------------------------------------
-if request_count >= 3:
-    st.warning("⚠️ Request limit reached (3 per user). Please wait before trying again.")
-    st.stop()
+user_prompt = st.text_area("✍️ Enter your prompt:", "Explain quantum computing in simple terms.")
 
 # -----------------------------------------------------
-# Sidebar: Model Settings and Future Works Section
+# Sidebar: Model Settings
 # -----------------------------------------------------
 st.sidebar.header("⚙️ Model Parameters")
 temperature = st.sidebar.slider("🎛 Temperature", 0.0, 1.5, 0.7)
 max_tokens = st.sidebar.slider("📏 Max Tokens", 50, 1000, 500)
 
+# -----------------------------------------------------
+# Sidebar Footer: Future Works Section
+# -----------------------------------------------------
 with st.sidebar:
     st.markdown("---")
     st.markdown("## 🔮 Future Works: 'Prompt Tester'")
@@ -68,14 +52,9 @@ with st.sidebar:
     """)
 
 # -----------------------------------------------------
-# User Input: Prompt
+# API Request Functions
 # -----------------------------------------------------
-user_prompt = st.text_area("✍️ Enter your prompt:", "Explain quantum computing in simple terms.")
 
-# -----------------------------------------------------
-# 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(
@@ -88,7 +67,6 @@ 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(
@@ -107,10 +85,6 @@ def get_gemini_response(prompt):
 # Generate Responses when Button is Clicked
 # -----------------------------------------------------
 if st.button("🚀 Generate Responses"):
-    request_count += 1
-    cookies["request_count"] = request_count  # Update cookie count
-    cookies.save()  # Persist changes
-
     with st.spinner("Fetching responses..."):
         openai_text = get_openai_response(user_prompt)
         gemini_text = get_gemini_response(user_prompt)