Update app.py

app.py

@@ -6,33 +6,46 @@ from dotenv import load_dotenv
 # Load environment variables from .env file
 load_dotenv()
 
-# Title of the app
+# Fetch API key from environment
+api_key = os.getenv("GROQ_API_KEY")
+
+# App Title
 st.title("Simple AI Agent with LLaMA 3.1")
 
-# Description
-st.markdown("This is an AI agent powered by the LLaMA 3.1 model and Groq API.")
+# App Description
+st.markdown("""
+This AI agent uses the LLaMA 3.1 model and Groq API to answer your queries.
+Please provide a question below to interact with the model.
+""")
 
-# Input for user queries
-user_input = st.text_input("Ask something:")
+# Warn if API key is missing
+if not api_key:
+    st.error("API key not found. Please ensure you have added a `.env` file with `GROQ_API_KEY`.")
 
-# Display response area
-if st.button("Get Response"):
-    # Fetch API key from .env file
-    api_key = os.getenv("GROQ_API_KEY")
-    
-    if api_key and user_input:
-        # Set up Groq client
-        client = Groq(api_key=api_key)
+# Input box for user query
+user_input = st.text_input("Enter your question:", placeholder="Ask something...")
 
+# Button to fetch response
+if st.button("Get Response"):
+    if user_input.strip():
         try:
-            # Send query to LLaMA model
+            # Initialize Groq client
+            client = Groq(api_key=api_key)
+            
+            # Query the AI model
             chat_completion = client.chat.completions.create(
                 messages=[{"role": "user", "content": user_input}],
                 model="llama3-8b-8192",
             )
-            # Display response
-            st.success(chat_completion.choices[0].message.content)
+            
+            # Display the response
+            st.success("AI Response:")
+            st.write(chat_completion.choices[0].message.content)
         except Exception as e:
             st.error(f"Error: {e}")
     else:
-        st.warning("Please ensure the API key is set in the .env file and ask a valid question.")
+        st.warning("Please enter a valid question!")
+
+# Footer
+st.markdown("---")
+st.markdown("Powered by **LLaMA 3.1** and **Groq API**.")