Update app.py

app.py

@@ -1,69 +1,47 @@
 import os
 import streamlit as st
-from groq import Groq
-
-# Initialize Groq client
-API_KEY = os.getenv("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")  # Ensure GROQ_API_KEY is set in your environment variables
-if not API_KEY:
-    st.error("API key is missing. Please set the GROQ_API_KEY environment variable.")
-    st.stop()
-
-client = Groq(api_key=API_KEY)
-
-# Streamlit app configuration
-st.set_page_config(page_title="AI Chatbot", page_icon="🤖", layout="wide")
-st.title("AI-Powered Chatbot with Groq")
-
-# Sidebar for user input
-st.sidebar.header("Chatbot Settings")
-chat_model = st.sidebar.selectbox("Select AI Model", ["llama-3.3-70b-versatile", "llama-2.0-50b-creative"])
-
-# Chat interface
-st.write("## Chat with AI")
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-
-# Display conversation history
-for msg in st.session_state.messages:
-    if msg["role"] == "user":
-        st.write(f"**You:** {msg['content']}")
-    else:
-        st.write(f"**AI:** {msg['content']}")
-
-# User input for the chatbot
-user_input = st.text_input("Enter your message:", key="user_input")
-
-# On submit, process the input
-if st.button("Send"):
-    if user_input:
-        # Add user input to the chat history
-        st.session_state.messages.append({"role": "user", "content": user_input})
-        
-        # Call Groq API
-        with st.spinner("Thinking..."):
-            try:
-                chat_completion = client.chat.completions.create(
-                    messages=st.session_state.messages,
-                    model=chat_model,
-                )
-                ai_response = chat_completion.choices[0].message.content
-
-                # Add AI response to the chat history
-                st.session_state.messages.append({"role": "assistant", "content": ai_response})
-
-                # Display the AI's response
-                st.write(f"**AI:** {ai_response}")
-            except Exception as e:
-                st.error(f"Error: {e}")
-    else:
-        st.warning("Please enter a message before sending.")
-
-# Clear conversation button
-if st.button("Clear Conversation"):
-    st.session_state.messages = []
-
-# Footer
-st.markdown("""
-    ---
-    Powered by [Groq](https://groq.com) | Deployed on [Hugging Face Spaces](https://huggingface.co/spaces)
-""")
+from groq import Client
+
+# Function to initialize Groq client
+def get_groq_client():
+    api_key = os.getenv("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")
+    if not api_key:
+        st.error("API key is missing. Please set the GROQ_API_KEY environment variable.")
+        return None
+    return Client(api_key=api_key)
+
+# Function to get chatbot response
+def get_response(client, user_input):
+    try:
+        response = client.chat.completions.create(
+            messages=[{"role": "user", "content": user_input}],
+            model="llama-3.3-70b-versatile"
+        )
+        return response['choices'][0]['message']['content']
+    except Exception as e:
+        st.error(f"Error: {e}")
+        return None
+
+# Streamlit UI
+def main():
+    st.title("AI Chatbot using Groq")
+
+    # Get the Groq client
+    client = get_groq_client()
+    if not client:
+        return
+
+    # User input
+    user_input = st.text_input("Ask a question:")
+    if st.button("Get Answer"):
+        if user_input:
+            with st.spinner("Thinking..."):
+                answer = get_response(client, user_input)
+            if answer:
+                st.success("Response:")
+                st.write(answer)
+        else:
+            st.warning("Please enter a message.")
+
+if __name__ == "__main__":
+    main()