Update app.py

app.py

@@ -40,14 +40,22 @@ def create_vector_db(text):
 
 # Function to query Groq API
 def query_groq_api(query, context, model="llama-3.3-70b-versatile"):
-    #url = "https://api.groq.com/openai/v1/chat/completions"
+    # Define the Groq API key
     GROQ_API_KEY = "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"
+    
+    # Optionally set it as an environment variable (not necessary in this case)
     os.environ["GROQ_API_KEY"] = GROQ_API_KEY
-    client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+
+    # API endpoint (Uncomment the URL)
+    url = "https://api.groq.com/openai/v1/chat/completions"
+    
+    # Headers for the API request
     headers = {
         "Content-Type": "application/json",
-        "Authorization": f"Bearer {os.getenv('GROQ_API_KEY')}",
+        "Authorization": f"Bearer {os.getenv('GROQ_API_KEY')}",  # Retrieve from environment
     }
+    
+    # Data to send to the API
     data = {
         "model": model,
         "messages": [
@@ -57,11 +65,18 @@ def query_groq_api(query, context, model="llama-3.3-70b-versatile"):
     }
 
     try:
+        # Send POST request to Groq API
         response = requests.post(url, headers=headers, json=data)
         response.raise_for_status()  # Raise an error for bad responses
+
+        # Get the API response content
         result = response.json()
+
+        # Extract the answer from the response
         return result.get("choices", [{}])[0].get("message", {}).get("content", "No response.")
+    
     except requests.exceptions.RequestException as e:
+        # Handle errors
         return f"Error: {e}"
 
 # Streamlit App