Update app.py

app.py

@@ -1,61 +1,65 @@
 import streamlit as st
+import openai
 
-# Set page config
+# Set Streamlit page configuration
 st.set_page_config(page_title="AI Prompt Enhancer", layout="centered")
 
-# Dark/Light Mode Toggle
-if "dark_mode" not in st.session_state:
-    st.session_state.dark_mode = False
-
-def toggle_mode():
-    st.session_state.dark_mode = not st.session_state.dark_mode
-
-# Apply Custom Styling
-dark_css = """
-    <style>
-        body { background-color: #121212; color: #ffffff; }
-        .stTextArea, .stButton>button { background-color: #1e1e1e; color: #ffffff; }
-        .stButton>button { border-radius: 10px; font-size: 16px; padding: 10px 20px; }
-    </style>
-"""
-
-light_css = """
-    <style>
-        body { background-color: #f5f5f5; color: #000000; }
-        .stTextArea, .stButton>button { background-color: #ffffff; color: #000000; }
-        .stButton>button { border-radius: 10px; font-size: 16px; padding: 10px 20px; }
-    </style>
-"""
-
-# Apply selected mode
-st.markdown(dark_css if st.session_state.dark_mode else light_css, unsafe_allow_html=True)
-
-# Header
+# OpenAI API Key (Store securely in Streamlit secrets)
+OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
+
+# Initialize OpenAI Client
+openai.api_key = OPENAI_API_KEY
+
+# 📌 Dictionary of Prompting Techniques
+prompt_techniques = {
+    "🔗 Chain of Thought (CoT)": "Breaks down reasoning into step-by-step logic.",
+    "📚 Few-Shot Learning": "Adds relevant examples for better AI understanding.",
+    "🧩 ReAct (Reasoning + Acting)": "Improves AI decision-making with iterative steps.",
+    "🌲 Tree of Thoughts (ToT)": "Generates multiple possible reasoning paths.",
+    "💡 Self-Consistency": "Reframes the prompt to generate diverse responses.",
+    "📑 HyDE (Hypothetical Doc Embeddings)": "Structures prompts for better retrieval-based outputs.",
+    "🔄 Least-to-Most Prompting": "Breaks complex tasks into smaller, manageable steps.",
+    "📊 Graph Prompting": "Optimizes structured data queries and relationships."
+}
+
+# 🎯 Title
 st.title("✨ AI Prompt Enhancer")
 
-# Dark Mode Toggle
-st.sidebar.button("🌙 Toggle Dark Mode" if not st.session_state.dark_mode else "☀️ Toggle Light Mode", on_click=toggle_mode)
+# 💡 Prompt Input Box
+user_prompt = st.text_area("💡 Prompt Idea Box", placeholder="Enter your prompt here...")
 
-# Prompt Input Box
-prompt = st.text_area("💡 Prompt Idea Box", placeholder="Type your prompt here...")
+# 📌 Dropdown for Selecting a Prompting Technique
+selected_technique = st.selectbox(
+    "🛠️ Choose a Prompting Technique",
+    options=list(prompt_techniques.keys()),
+    format_func=lambda x: f"{x} - {prompt_techniques[x]}"  # Display name + description
+)
 
-# Category Selection
-st.subheader("🎛️ Select an Enhancement Type")
+# 🚀 Enhance Prompt Button
+if st.button("Enhance Prompt"):
+    if user_prompt.strip() == "":
+        st.warning("⚠️ Please enter a prompt before enhancing.")
+    else:
+        # Generate enhanced prompt using OpenAI API
+        try:
+            response = openai.ChatCompletion.create(
+                model="gpt-4o",
+                messages=[
+                    {"role": "system", "content": f"Enhance this prompt using {selected_technique}."},
+                    {"role": "user", "content": user_prompt}
+                ],
+                temperature=0.7,
+                max_tokens=500
+            )
+            enhanced_prompt = response["choices"][0]["message"]["content"]
 
-categories = {
-    "📜 Copywriting": ["Ad Copy", "Blog Intro", "Product Description"],
-    "💻 Code Generation": ["Python", "SQL", "Automation Scripts"],
-    "📧 Email & Outreach": ["Cold Emails", "Sales Pitches"],
-    "📝 Academic & Research": ["Essays", "Research Summaries"],
-    "🔍 SEO Optimization": ["Keyword Rewrites", "Meta Descriptions"],
-    "🎭 Creative Writing": ["Storytelling", "Poetry", "Social Media Posts"]
-}
+            # 📌 Display Enhanced Prompt
+            st.subheader("🔹 Enhanced Prompt:")
+            st.text_area(" ", value=enhanced_prompt, height=150)
 
-selected_category = st.selectbox("Choose a category:", list(categories.keys()))
+            # 📋 Copy Button
+            st.button("📋 Copy Prompt", on_click=lambda: st.experimental_set_query_params(prompt=enhanced_prompt))
 
-# Display subcategories as buttons
-selected_subcategory = st.radio("Choose an enhancement type:", categories[selected_category])
+        except Exception as e:
+            st.error(f"❌ Error: {str(e)}")
 
-# Enhance Prompt Button
-if st.button("🚀 Enhance Prompt"):
-    st.write(f"🔹 **Enhanced Prompt for {selected_subcategory}:** [Prompt will be generated here]")