Update app.py

app.py

@@ -1,48 +1,56 @@
 import streamlit as st
-import openai
+import requests
 import os
 
-# Set OpenAI API Key
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Load Groq API key
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
-# Check API key
-if not openai.api_key:
-    st.error("❌ OPENAI_API_KEY not found. Add it in Settings → Secrets in your Hugging Face Space.")
+if not GROQ_API_KEY:
+    st.error("❌ GROQ_API_KEY not found! Please add it in Settings → Secrets.")
     st.stop()
 
-# Page UI
-st.set_page_config(page_title="AI Quest Generator", page_icon="⚔️")
-st.title("⚔️ AI-Powered Quest Generator")
-st.markdown("Choose your character class and generate a fantasy adventure!")
+# 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!")
 
-# Character selection
-classes = ["🛡️ Warrior", "🧙 Mage", "🗡️ Thief", "🏹 Ranger", "🐉 Dragonborn"]
-selected_class = st.selectbox("Choose your class:", classes)
+# Character classes
+classes = ["🔥 Fire Knight", "❄️ Ice Mage", "🩸 Shadow Assassin", "⚡ Storm Archer", "🐉 Dragon Tamer"]
+selected_class = st.selectbox("Select Your Warrior:", classes)
+
+if st.button("⚔️ Enter the Arena"):
+    with st.spinner("Groq is writing your battle..."):
 
-# Generate button
-if st.button("🎲 Generate My Quest"):
-    with st.spinner("Summoning your quest..."):
         prompt = f"""
-        You are a game master. Create an exciting fantasy quest for a player who is a {selected_class}.
+        Write a cinematic fantasy battle scene for a hero who is a {selected_class}.
         Include:
-        - Quest title
-        - Setting description
-        - Mission objective
-        - Main enemy
-        - Twist or surprise
-        Write in an engaging story style.
+        - 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 = openai.ChatCompletion.create(
-                model="gpt-4o",
-                messages=[{"role": "user", "content": prompt}],
-                temperature=0.9,
-                max_tokens=700
-            )
-            quest = response.choices[0].message.content
-            st.success("🧭 Quest Ready!")
-            st.markdown(f"```markdown\n{quest}\n```")
+            res = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=json_data)
+            output = res.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}")
+            st.error(f"❌ Error: {e}")