Update app.py

app.py

@@ -2,24 +2,26 @@ import streamlit as st
 import requests
 import os
 
-# Load Groq API key
+# Load Groq API key from Hugging Face Secrets
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
+# Check API key
 if not GROQ_API_KEY:
-    st.error("❌ GROQ_API_KEY not found! Please add it in Settings → Secrets.")
+    st.error("❌ GROQ_API_KEY not found! Add it in Settings → Secrets.")
     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!")
+st.markdown("Choose your fighter and generate an epic AI battle scene powered by LLaMA 3!")
 
-# Character classes
+# Character selection
 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("Groq is writing your battle..."):
+    with st.spinner("Summoning your battle..."):
 
         prompt = f"""
         Write a cinematic fantasy battle scene for a hero who is a {selected_class}.
@@ -38,19 +40,21 @@ if st.button("⚔️ Enter the Arena"):
 
         json_data = {
             "model": "llama3-8b-8192",
-            "messages": [
-                {"role": "user", "content": prompt}
-            ],
+            "messages": [{"role": "user", "content": prompt}],
             "temperature": 0.9,
             "max_tokens": 700,
             "top_p": 1
         }
 
         try:
-            res = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=json_data)
-            output = res.json()['choices'][0]['message']['content']
+            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.")