Update app.py

app.py

@@ -1,40 +1,29 @@
 import streamlit as st
-from transformers import pipeline
-from huggingface_hub import login
-import os
+import random
 
-HF_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
-login(HF_TOKEN)
+# Page title
+st.set_page_config(page_title="🎲 Dice Roller Story Bot", page_icon="🎲")
+st.title("🎲 Dice Roller + Mini Story Bot")
+st.markdown("Click the button to roll a dice and see what happens in your story!")
 
-st.title("🤖📟 Arduino Expert Chatbot")
-st.markdown("Ask anything about Arduino: code, circuits, projects!")
+# Define dice roll outcomes
+story_outcomes = {
+    1: "🧟 You encountered a zombie and ran for your life!",
+    2: "💰 You found a hidden treasure under a tree!",
+    3: "🐉 A dragon appeared... but it just wanted a hug.",
+    4: "🛡️ You discovered an ancient shield with magical powers!",
+    5: "🍄 You ate a glowing mushroom and now you can talk to animals.",
+    6: "🕳️ You fell into a secret tunnel leading to a lost kingdom!"
+}
 
-@st.cache_resource
-def load_model():
-    try:
-        model = pipeline(
-            "text-generation",
-            model="tiiuae/falcon-7b-instruct",  # lightweight, non-gated
-            token=HF_TOKEN
-        )
-        return model
-    except Exception as e:
-        print(f"Model load failed: {e}")
-        st.error("❌ Failed to load model.")
-        return pipeline("text-generation", model="gpt2")
+# Roll the dice
+if st.button("Roll the Dice 🎲"):
+    roll = random.randint(1, 6)
+    st.subheader(f"You rolled a **{roll}**!")
+    st.success(story_outcomes[roll])
+else:
+    st.info("Ready to roll? Click the button above!")
 
-model = load_model()
-
-query = st.text_area("Ask your Arduino question here 👇", height=150)
-
-if st.button("Get Answer"):
-    if query.strip():
-        with st.spinner("Thinking... 🤖"):
-            try:
-                prompt = f"<s>[INST] {query} [/INST]"
-                response = model(prompt, max_new_tokens=512, do_sample=True, temperature=0.7)
-                st.success(response[0]['generated_text'])
-            except Exception as e:
-                st.error(f"❌ Error generating response: {e}")
-    else:
-        st.warning("Please enter a valid question.")
+# Footer
+st.markdown("---")
+st.caption("Made with ❤️ using Streamlit and Hugging Face Spaces")