Update app.py

app.py

@@ -9,76 +9,87 @@ OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
 GEMINI_API_KEY = st.secrets["GEMINI_API_KEY"]
 
 # -----------------------------------------------------
-# Initialize OpenAI Client (New Method)
+# Initialize OpenAI & Gemini
 # -----------------------------------------------------
-client = OpenAI(api_key=OPENAI_API_KEY)  # New OpenAI API usage
-
-# Initialize Gemini Model
+client = OpenAI(api_key=OPENAI_API_KEY)  # OpenAI client
 genai.configure(api_key=GEMINI_API_KEY)
-gemini_model = genai.GenerativeModel("gemini-2.0-flash")
+gemini_model = genai.GenerativeModel("gemini-pro")
 
 # -----------------------------------------------------
 # Configure Streamlit Page
 # -----------------------------------------------------
-st.set_page_config(page_title="AI Model Comparison", layout="wide")
-st.title("🔍 Compare AI Model Responses")
-st.subheader("Enter a prompt below and see how different AI models respond.")
+st.set_page_config(page_title="AI Prompt Tester", layout="wide")
+st.title("🧪 AI Prompt Tester")
+st.subheader("Fine-tune your prompts & compare LLM responses.")
 
 # -----------------------------------------------------
-# Create a 2-Column Layout for OpenAI and Gemini
+# User Input: Prompt
 # -----------------------------------------------------
-col1, col2 = st.columns(2)
+user_prompt = st.text_area("✍️ Enter your prompt:", "Explain quantum computing in simple terms.")
 
 # -----------------------------------------------------
-# Chatbox for User Input
+# 🔥 New Feature 1: Temperature & Max Tokens Sliders
 # -----------------------------------------------------
-user_prompt = st.text_area("Enter your prompt:", "Explain quantum computing in simple terms.")
+st.sidebar.header("⚙️ Model Parameters")
+temperature = st.sidebar.slider("🎛 Temperature (0 = predictable, 1 = creative)", 0.0, 1.0, 0.7)
+max_tokens = st.sidebar.slider("📏 Max Tokens (limits response length)", 50, 1000, 500)
 
 # -----------------------------------------------------
-# Generate Responses from OpenAI and Gemini
+# 🔄 New Feature 2: "Reword My Prompt"
+# -----------------------------------------------------
+if st.button("♻️ Reword My Prompt"):
+    try:
+        reworded_prompt = client.chat.completions.create(
+            model="gpt-4o",
+            messages=[{"role": "user", "content": f"Reword this prompt differently: {user_prompt}"}]
+        )
+        st.write("🔄 **Alternative Prompt:**", reworded_prompt.choices[0].message.content)
+    except Exception as e:
+        st.error(f"Error in rewording: {e}")
+
 # -----------------------------------------------------
-if st.button("Generate Responses"):
+# Generate Responses from OpenAI & Gemini
+# -----------------------------------------------------
+if st.button("🚀 Generate Responses"):
     with st.spinner("Fetching responses..."):
+        
         # --- OpenAI GPT-4o ---
         try:
-            completion = client.chat.completions.create(
+            openai_response = client.chat.completions.create(
                 model="gpt-4o",
-                messages=[{"role": "user", "content": user_prompt}]
+                messages=[{"role": "user", "content": user_prompt}],
+                temperature=temperature,
+                max_tokens=max_tokens
             )
-            openai_text = completion.choices[0].message.content
+            openai_text = openai_response.choices[0].message.content
         except Exception as e:
-            openai_text = f"OpenAI error: {e}"
+            openai_text = f"❌ OpenAI error: {e}"
         
         # --- Gemini AI ---
         try:
             response = gemini_model.generate_content(
                 user_prompt,
                 generation_config=genai.types.GenerationConfig(
-                    temperature=0.0,          # Ensures deterministic output
-                    max_output_tokens=500,    # Limits the response length to 500 tokens
-                    candidate_count=1         # Generates only one candidate
+                    temperature=temperature,
+                    max_output_tokens=max_tokens,
+                    candidate_count=1
                 )
             )
-            gemini_text = response.text  # Extract text output
+            gemini_text = response.text
         except Exception as e:
-            gemini_text = f"Gemini error: {e}"
+            gemini_text = f"❌ Gemini error: {e}"
         
         # -----------------------------------------------------
-        # Display responses in respective columns
+        # 🆕 New Feature 3: Display responses in Expanders
         # -----------------------------------------------------
-        with col1:
-            st.markdown("### OpenAI GPT-4o")
-            st.write(openai_text)
-        
-        with col2:
-            st.markdown("### Gemini AI")
-            st.write(gemini_text)
-        
+        with st.expander("🧠 OpenAI GPT-4o Response"):
+            st.markdown(f"**GPT-4o says:**\n{openai_text}")
+
+        with st.expander("🌍 Gemini AI Response"):
+            st.markdown(f"**Gemini says:**\n{gemini_text}")
+
         # -----------------------------------------------------
-        # Placeholders for future integration of DeepSeek and Ollama
+        # 🆕 New Feature 4: User Feedback Rating
         # -----------------------------------------------------
-        st.markdown("### DeepSeek AI")
-        st.write("DeepSeek integration will be added later.")
-        
-        st.markdown("### Ollama")
-        st.write("Ollama integration will be added later..")
+        rating = st.radio("📊 Which model gave the best response?", ["OpenAI GPT-4o", "Gemini"])
+        st.success(f"✅ You selected: {rating}")