Update app.py

app.py

@@ -1,76 +1,41 @@
 import streamlit as st
-from pyngrok import ngrok
+from groq import Groq
+import os
 
-# Kill existing tunnels to avoid conflicts
-ngrok.kill()
+# API key from secrets
+client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
-# Connect to port 8501 (default for Streamlit)
-public_url = ngrok.connect(8501)
-st.write(f"Ngrok tunnel URL: {public_url}")
+st.set_page_config(page_title="AI Chatbot", page_icon="🤖")
 
-# Your main Streamlit app code
 st.title("🤖 AI Chatbot (Groq)")
-st.write("Fast AI chatbot 🚀")
+st.write("Powered by Groq ⚡")
 
-# Initialize chat messages in session state
+# Chat memory
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# Display chat messages
+# Show messages
 for msg in st.session_state.messages:
     with st.chat_message(msg["role"]):
         st.markdown(msg["content"])
 
-# User input
+# Input
 user_input = st.chat_input("Type your message...")
 
 if user_input:
-    # Append user message
     st.session_state.messages.append({"role": "user", "content": user_input})
+
     with st.chat_message("user"):
         st.markdown(user_input)
 
-    # Generate response
     with st.chat_message("assistant"):
         with st.spinner("Thinking..."):
             response = client.chat.completions.create(
-                model="mixtral-8x7b-32768",  # or your preferred model
+                model="mixtral-8x7b-32768",
                 messages=st.session_state.messages,
                 max_tokens=500
             )
             reply = response.choices[0].message.content
             st.markdown(reply)
-            # Append assistant response
-            st.session_state.messages.append({"role": "assistant", "content": reply})
-
-# Connect ngrok to port 8501 (Streamlit default port)
-from pyngrok import ngrok
-
-# Disconnect all existing tunnels
-public_url = ngrok.connect(8501)
-st.write(f"👉 Your app is live at: {public_url}")
-"""
-
-# Save the code to a Python file
-with open("streamlit_app.py", "w") as f:
-    f.write(app_code)
-
-# Step 4: Run the Streamlit app
-!streamlit run streamlit_app.py &>/dev/null &
-
-import json
-import time
-from pyngrok import ngrok
-
-
-# Connect to port 8501
-public_url = ngrok.connect(8501)
-
-# Wait a moment to ensure the tunnel is established
-time.sleep(2)
 
-# Get the public URL
-tunnels = ngrok.get_tunnels()
-for tunnel in tunnels:
-    if tunnel.public_url:
-        print(f"Your Streamlit app is live at: {tunnel.public_url}")
+    st.session_state.messages.append({"role": "assistant", "content": reply})