Update app.py

app.py

@@ -1,60 +1,61 @@
-import streamlit as st 
+import streamlit as st
 import requests
 import os
 
-# Load Groq API key from Hugging Face Secrets
+# Set up Streamlit UI
+st.set_page_config(page_title="Groq Battle Game", page_icon="⚔️")
+st.title("⚔️ Groq Battle Game")
+st.markdown("Welcome, warrior! Choose your hero and enter the arena!")
+
+# Get Groq API key from Hugging Face Secrets
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
-# Check API key
+# Show error if API key not found
 if not GROQ_API_KEY:
-    st.error("❌ GROQ_API_KEY not found! Add it in Settings → Secrets.")
+    st.error("❌ GROQ_API_KEY is missing. Go to your Hugging Face Space → Settings → Secrets and add it.")
     st.stop()
 
-# App UI
-st.set_page_config(page_title="Groq Battle Arena", page_icon="⚔️")
-st.title("⚔️ Groq Battle Arena")
-st.markdown("Choose your fighter and generate an epic AI battle scene powered by LLaMA 3!")
-
-# Character selection
+# Hero Class Dropdown
 classes = ["🔥 Fire Knight", "❄️ Ice Mage", "🩸 Shadow Assassin", "⚡ Storm Archer", "🐉 Dragon Tamer"]
-selected_class = st.selectbox("Select Your Warrior:", classes)
-
-# Generate battle on button click
-if st.button("⚔️ Enter the Arena"):
-    with st.spinner("Summoning your battle..."):
-
-        prompt = f"""
-        Write a cinematic fantasy battle scene for a hero who is a {selected_class}.
-        Include:
-        - Setting of the battle
-        - Description of the enemy
-        - Special powers used
-        - Epic conclusion
-        Write like a fantasy novel, in storytelling tone.
-        """
-
-        headers = {
-            "Authorization": f"Bearer {GROQ_API_KEY}",
-            "Content-Type": "application/json"
-        }
-
-        json_data = {
-            "model": "llama3-8b-8192",
-            "messages": [{"role": "user", "content": prompt}],
-            "temperature": 0.9,
-            "max_tokens": 700,
-            "top_p": 1
-        }
-
-        try:
-            response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=json_data)
-            output = response.json()['choices'][0]['message']['content']
-            st.success("⚔️ Battle Generated!")
-            st.markdown(f"```markdown\n{output}\n```")
-
-        except Exception as e:
-            st.error(f"❌ Error: {e}")
-
-# Optional bottom message to help Hugging Face recognize app
-if __name__ == "__main__":
-    st.write("✅ App initialized.")
+hero = st.selectbox("Choose your hero class:", classes)
+
+# Button to generate story
+if st.button("Enter Battle"):
+    st.info("⚔️ Generating your battle scene using Groq...")
+
+    # Prompt for Groq API
+    prompt = f"""
+    Write a dramatic fantasy battle story where the hero is a {hero}.
+    The story should include:
+    - An intense setting
+    - A powerful enemy
+    - Use of special abilities
+    - A heroic or surprising ending
+    Write it like a fantasy novel, with vivid descriptions.
+    """
+
+    headers = {
+        "Authorization": f"Bearer {GROQ_API_KEY}",
+        "Content-Type": "application/json"
+    }
+
+    data = {
+        "model": "llama3-8b-8192",
+        "messages": [{"role": "user", "content": prompt}],
+        "temperature": 0.9,
+        "max_tokens": 700
+    }
+
+    try:
+        response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=data)
+        result = response.json()
+        story = result['choices'][0]['message']['content']
+        st.success("🎉 Your battle story is ready!")
+        st.markdown(f"```\n{story}\n```")
+
+    except Exception as e:
+        st.error(f"🚨 Error: {e}")
+
+# Footer
+st.markdown("---")
+st.markdown("Made with ❤️ using Groq + Streamlit on Hugging Face Spaces")