Update app.py

app.py

@@ -2,15 +2,15 @@ import streamlit as st
 import google.generativeai as genai
 import time
 
-# Hugging Face Streamlit UI Configuration
-st.set_page_config(page_title="PromptLab", layout="wide")
+# ✅ Streamlit Page Configuration for Hugging Face Spaces
+st.set_page_config(page_title="PromptLab - AI Prompt Enhancer", layout="wide")
 st.title("⚡ PromptLab - AI Prompt Enhancer")
 
-# Retrieve the API key from Streamlit secrets
+# ✅ Retrieve the API key from Hugging Face secrets
 GOOGLE_API_KEY = st.secrets["GEMINI_API_KEY"]
 genai.configure(api_key=GOOGLE_API_KEY)
 
-# Define Shinobi and Raikage prompts
+# ✅ Define Shinobi and Raikage Prompt Frameworks
 SHINOBI_PROMPT = """You are an advanced prompt enhancer, specializing in creating structured, high-clarity prompts that optimize LLM performance.  
 Your task is to refine a given prompt using the **Shinobi framework**, ensuring the following principles:  
 
@@ -45,33 +45,37 @@ Your task is to refine a given prompt using the **Raikage framework**, ensuring
 **Enhanced Raikage Prompt:**  
 """
 
-# Streamlit Layout
-mode = st.radio("🔥 Choose a mode:", ["🌀 Shinobi", "⚡ Raikage"], horizontal=True)
+# ✅ Streamlit UI Components
+st.subheader("🛠️ Choose Your Enhancement Mode:")
+mode = st.radio("Select a mode:", ["🌀 Shinobi", "⚡ Raikage"], horizontal=True)
+
 user_prompt = st.text_area("✍️ Enter your prompt:", height=150)
 
-# Button to enhance prompt
+# ✅ Button to Enhance Prompt
 if st.button("🚀 Enhance Prompt"):
     if not user_prompt.strip():
         st.warning("⚠️ Please enter a prompt before enhancing!")
     else:
         with st.spinner("⚡ Enhancing your prompt... Please wait"):
-            time.sleep(1)  # Simulate slight delay for better UI response
+            time.sleep(1)  # 🔄 Smooth UI transition
             
-            # Select the correct system prompt
+            # Select the appropriate enhancement framework
             if mode == "🌀 Shinobi":
                 full_prompt = SHINOBI_PROMPT.format(user_prompt=user_prompt)
             else:
                 full_prompt = RAIKAGE_PROMPT.format(user_prompt=user_prompt)
             
-            # Initialize Gemini Model & Call API
+            # ✅ Call Gemini API to Enhance the Prompt
             try:
                 model = genai.GenerativeModel('gemini-pro')
                 response = model.generate_content(full_prompt)
                 
-                # Display Output
+                # ✅ Display Enhanced Prompt
                 st.subheader("✨ Enhanced Prompt:")
-                st.write(response.text)
+                st.text_area("", response.text, height=200)  # Read-only box
                 
+                # ✅ Copy to Clipboard Button
+                st.code(response.text, language="markdown")
+
             except Exception as e:
                 st.error(f"❌ API Error: {e}")
-