Create app.py

app.py

@@ -0,0 +1,38 @@
+import streamlit as st
+import os
+from groq import Groq
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
+
+# Title of the app
+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.")
+
+# Input for user queries
+user_input = st.text_input("Ask something:")
+
+# 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)
+
+        try:
+            # Send query to LLaMA 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)
+        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.")