Update app.py

app.py

@@ -50,15 +50,29 @@ import os
 load_dotenv()
 
 # Get API key from env
-api_key = os.getenv("FIREWORKS_API_KEY")
-if not api_key:
-    raise EnvironmentError("Please set FIREWORKS_API_KEY in your .env or Space secrets")
-
-# Initialize OpenAI client (pointing to Fireworks API)
-client = OpenAI(
-    base_url="https://api.fireworks.ai/inference/v1/chat/completions",
-    api_key=api_key,
-)
+url = "https://api.fireworks.ai/inference/v1/chat/completions"
+payload = {
+  "model": "accounts/fireworks/models/llama-v3p1-405b-instruct",
+  "max_tokens": 16384,
+  "top_p": 1,
+  "top_k": 40,
+  "presence_penalty": 0,
+  "frequency_penalty": 0,
+  "temperature": 0.6,
+  "messages": [
+    {
+      "role": "user",
+      "content": "Hello, how are you?"
+    }
+  ]
+}
+headers = {
+  "Accept": "application/json",
+  "Content-Type": "application/json",
+  "Authorization": "Bearer <FIREWORKS_API_KEY>"
+}
+requests.request("POST", url, headers=headers, data=json.dumps(payload))
+
 
 # Make a test call
 response = client.chat.completions.create(