Create app.py

app.py

@@ -0,0 +1,116 @@
+import streamlit as st
+from transformers import pipeline
+import torch
+import random
+
+# Set page config
+st.set_page_config(
+    page_title="🧠 Brainrot Chat",
+    page_icon="🧠"
+)
+
+# Initialize session state for chat history if it doesn't exist
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Fun loading messages
+LOADING_MESSAGES = [
+    "Rotting brain cells... 🧠",
+    "Downloading TikTok wisdom... 📱",
+    "Absorbing internet culture... 🌐",
+    "Loading peak brainrot... 🤪",
+    "Gathering viral knowledge... 🦠"
+]
+
+def load_qa_pipeline():
+    """Load the question-answering pipeline"""
+    if "qa_pipeline" not in st.session_state:
+        with st.spinner(random.choice(LOADING_MESSAGES)):
+            st.session_state.qa_pipeline = pipeline(
+                "question-answering",
+                model="CallmeKaito/llama-3.1-8b-it-brainrot",
+                device=0 if torch.cuda.is_available() else -1
+            )
+
+def get_answer(question, context):
+    """Get answer from the model"""
+    try:
+        result = st.session_state.qa_pipeline(
+            question=question,
+            context=context
+        )
+        return result['answer']
+    except Exception as e:
+        return f"Brain too rotted, can't compute 🤯 Error: {str(e)}"
+
+# Main UI with themed styling
+st.markdown("""
+    <style>
+    .big-font {
+        font-size: 40px !important;
+        font-weight: bold;
+    }
+    </style>
+    <p class="big-font">🧠 Brainrot Chat 🤪</p>
+    """, unsafe_allow_html=True)
+
+st.markdown("""
+    Welcome to the Brainrot Zone! This chatbot has been trained on peak TikTok content.
+    Prepare for some absolutely unhinged responses! 
+    
+    ⚠️ Remember: This is for entertainment only - don't take anything too seriously! ⚠️
+    """)
+
+# Load the model
+load_qa_pipeline()
+
+# Example prompts to help users
+EXAMPLE_CONTEXTS = [
+    "When the rizz is immaculate but she's literally so real for what fr fr no cap bussin",
+    "POV: You're chronically online and everything is giving main character energy",
+    "It's giving slay queen energy with a side of based behavior ngl",
+]
+
+# Context input with example
+context = st.text_area(
+    "Drop your TikTok wisdom here:",
+    placeholder=random.choice(EXAMPLE_CONTEXTS),
+    height=100
+)
+
+# Display chat history
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.write(message["content"])
+
+# User input
+if question := st.chat_input("Ask something absolutely unhinged..."):
+    # Display user message
+    with st.chat_message("user"):
+        st.write(question)
+    st.session_state.messages.append({"role": "user", "content": question})
+    
+    # Generate and display assistant response
+    if context:
+        with st.chat_message("assistant"):
+            answer = get_answer(question, context)
+            st.write(f"{answer} {random.choice(['💅', '😌', '💁', '🤪', '✨'])}")
+        st.session_state.messages.append({"role": "assistant", "content": answer})
+    else:
+        with st.chat_message("assistant"):
+            st.write("Bestie, I need some context to work with! Drop some TikTok wisdom above! ✨")
+        st.session_state.messages.append(
+            {"role": "assistant", "content": "Bestie, I need some context to work with! Drop some TikTok wisdom above! ✨"}
+        )
+
+# Clear chat button with themed text
+if st.button("Reset the Brainrot 🧠"):
+    st.session_state.messages = []
+    st.experimental_rerun()
+
+# Footer
+st.markdown("""
+    ---
+    *This chatbot is living its best life with maximum brainrot energy. 
+    Any weird responses are just part of the aesthetic* ✨
+    """)