Update app.py

app.py

@@ -1,72 +1,77 @@
 import streamlit as st
 import google.generativeai as genai
+import time
 
-# 🔑 Load API Key from Streamlit Secrets
+# Hugging Face Streamlit UI Configuration
+st.set_page_config(page_title="PromptLab", layout="wide")
+st.title("⚡ PromptLab - AI Prompt Enhancer")
+
+# Retrieve the API key from Streamlit secrets
 GOOGLE_API_KEY = st.secrets["GEMINI_API_KEY"]
 genai.configure(api_key=GOOGLE_API_KEY)
 
-# 🎨 Streamlit UI Setup
-st.title("⚡ PromptLab – AI-Powered Prompt Enhancer")
-st.write("Enhance your prompts with **Shinobi** (Structured) or **Raikage** (Execution-Focused)")
+# Define Shinobi and Raikage prompts
+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:  
+
+✅ **Concise & High-Density Prompting** → Remove fluff, keeping instructions clear and actionable (~250 words max).  
+✅ **Explicit Role Definition** → Assign a role to the AI for better contextual grounding.  
+✅ **Step-by-Step Clarity** → Break the task into structured sections, avoiding ambiguity.  
+✅ **Defined Output Format** → Specify the response format (JSON, CSV, list, structured text, etc.).  
+✅ **Zero Conflicting Instructions** → Ensure clarity in constraints (e.g., avoid “simple yet comprehensive”).  
+✅ **Optional: One-Shot Example** → Add a single example where relevant to guide the AI.  
+
+### **Enhance the following prompt using Shinobi principles:**  
+**Original Prompt:**  
+{user_prompt}  
+
+**Enhanced Shinobi Prompt:**  
+"""
+
+RAIKAGE_PROMPT = """You are an elite AI strategist, specializing in designing execution-focused prompts that maximize LLM efficiency.  
+Your task is to refine a given prompt using the **Raikage framework**, ensuring the following principles:  
 
-# 🎭 Mode Selection
-mode = st.radio("Select Mode:", ["🌀 Shinobi", "⚡ Raikage"], horizontal=True)
+✅ **Precision & Depth** → Ensure expert-level guidance, reducing vagueness and ambiguity.  
+✅ **Context & Execution Approach** → Include a structured methodology to solve the problem.  
+✅ **Defined Output Format** → Specify exact structure (JSON, formatted text, markdown, tables, or code blocks).  
+✅ **Edge Case Handling & Constraints** → Account for potential failures and model limitations.  
+✅ **Optional: Few-Shot Prompting** → If beneficial, provide 1-2 high-quality examples for refinement.  
+✅ **Complies with External Factors** → Adhere to best practices (e.g., ethical scraping, security policies).  
 
-# ✍️ User Input for Prompt
-user_prompt = st.text_area("Enter your prompt:")
+### **Enhance the following prompt using Raikage principles:**  
+**Original Prompt:**  
+{user_prompt}  
 
-# 🚀 Generate Enhanced Prompt
-if st.button("Enhance Prompt"):
+**Enhanced Raikage Prompt:**  
+"""
+
+# Streamlit Layout
+mode = st.radio("🔥 Choose a mode:", ["🌀 Shinobi", "⚡ Raikage"], horizontal=True)
+user_prompt = st.text_area("✍️ Enter your prompt:", height=150)
+
+# Button to enhance prompt
+if st.button("🚀 Enhance Prompt"):
     if not user_prompt.strip():
-        st.warning("⚠️ Please enter a prompt before enhancing.")
+        st.warning("⚠️ Please enter a prompt before enhancing!")
     else:
-        with st.spinner("Enhancing your prompt... ⚡"):
-            # 🛠️ Apply Shinobi or Raikage Framework
+        with st.spinner("⚡ Enhancing your prompt... Please wait"):
+            time.sleep(1)  # Simulate slight delay for better UI response
+            
+            # Select the correct system prompt
             if mode == "🌀 Shinobi":
-                structured_prompt = f"""
-                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:
-                
-                ✅ **Concise & High-Density Prompting** → Remove fluff, keeping instructions clear and actionable.
-                ✅ **Explicit Role Definition** → Assign a role to the AI for better contextual grounding.
-                ✅ **Step-by-Step Clarity** → Break the task into structured sections.
-                ✅ **Defined Output Format** → Specify the response format.
-                
-                ### **Enhance the following prompt using Shinobi principles:**
-                
-                **Original Prompt:**  
-                {user_prompt}
-                
-                **Enhanced Shinobi Prompt:**  
-                """
+                full_prompt = SHINOBI_PROMPT.format(user_prompt=user_prompt)
             else:
-                structured_prompt = f"""
-                You are an elite AI strategist, specializing in designing execution-focused prompts that maximize LLM efficiency.
-                Your task is to refine a given prompt using the **Raikage framework**, ensuring:
-                
-                ✅ **Precision & Depth** → Ensure expert-level guidance, reducing vagueness.
-                ✅ **Context & Execution Approach** → Include a structured methodology.
-                ✅ **Defined Output Format** → Specify exact structure (JSON, markdown, tables, etc.).
-                ✅ **Edge Case Handling & Constraints** → Account for potential failures.
-                
-                ### **Enhance the following prompt using Raikage principles:**
+                full_prompt = RAIKAGE_PROMPT.format(user_prompt=user_prompt)
+            
+            # Initialize Gemini Model & Call API
+            try:
+                model = genai.GenerativeModel('gemini-pro')
+                response = model.generate_content(full_prompt)
                 
-                **Original Prompt:**  
-                {user_prompt}
+                # Display Output
+                st.subheader("✨ Enhanced Prompt:")
+                st.write(response.text)
                 
-                **Enhanced Raikage Prompt:**  
-                """
-
-            # 🧠 Call Gemini API
-            model = genai.GenerativeModel('gemini-pro')
-            response = model.generate_content(structured_prompt)
-
-            # 📝 Display Output
-            enhanced_prompt = response.text.strip()
-            st.subheader("🔹 Enhanced Prompt:")
-            st.code(enhanced_prompt, language="markdown")
-
-            # 📋 Copy Button
-            st.button("📋 Copy to Clipboard", on_click=lambda: st.session_state.update({"copied_text": enhanced_prompt}))
-
+            except Exception as e:
+                st.error(f"❌ API Error: {e}")