Update app.py

app.py

@@ -4,17 +4,18 @@ from together import Together
 
 @cl.on_message
 async def main(message: str):
-    # Read your Together API key from environment
+    # Read your Together API key from the environment
     together_api_key = os.getenv("TOGETHER_API_KEY", "")
     if not together_api_key:
         await cl.Message(
-            content="Error: No Together API key found. Please set TOGETHER_API_KEY."
+            content="Error: No Together API key found. Please set TOGETHER_API_KEY in your Space secrets."
         ).send()
         return
 
-    # Initialize the Together client with your API key
+    # Initialize the Together client
     client = Together(api_key=together_api_key)
 
+    # Construct messages
     messages = [
         {
             "role": "user",
@@ -23,15 +24,19 @@ async def main(message: str):
     ]
 
     try:
-        # Call the desired model
+        # Call the model
         completion = client.chat.completions.create(
             model="meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
             messages=messages,
             max_tokens=500
         )
-        # Extract response
+
+        # Extract the response text
         response = completion.choices[0].message.content
-        # Send Chainlit message
+
+        # Send the response to Chainlit
         await cl.Message(content=response).send()
+
     except Exception as e:
+        # Send any error info to Chainlit
         await cl.Message(content=f"Error: {str(e)}").send()