Update app.py

app.py

@@ -1,27 +1,24 @@
-import os
 import chainlit as cl
 from together import Together
 
+# Hardcoded Together API key for testing (replace with your key)
+TOGETHER_API_KEY = "4e381cebe7224f3da54cae6e54d3fdd3ee00b6ac160fc29cea66fbe48a0be3d1"
+
 @cl.on_message
 async def main(message: str):
-    # Read your Together API key from the environment
-    together_api_key = os.getenv("TOGETHER_API_KEY", "")
+    # Use the hardcoded key directly
+    together_api_key = TOGETHER_API_KEY
     if not together_api_key:
         await cl.Message(
-            content="Error: No Together API key found. Please set TOGETHER_API_KEY in your Space secrets."
+            content="Error: No Together API key provided."
         ).send()
         return
 
     # Initialize the Together client
     client = Together(api_key=together_api_key)
 
-    # Construct messages
-    messages = [
-        {
-            "role": "user",
-            "content": message
-        }
-    ]
+    # Construct messages from the user input
+    messages = [{"role": "user", "content": message}]
 
     try:
         # Call the model
@@ -30,13 +27,10 @@ async def main(message: str):
             messages=messages,
             max_tokens=500
         )
-
-        # Extract the response text
+        # Extract the response
         response = completion.choices[0].message.content
-
-        # Send the response to Chainlit
+        # Send the response via Chainlit
         await cl.Message(content=response).send()
-
     except Exception as e:
-        # Send any error info to Chainlit
+        # If there is an error, send the error message
         await cl.Message(content=f"Error: {str(e)}").send()