Update app.py

app.py

@@ -30,6 +30,20 @@ def call_groq_api(prompt, model="llama3-70b-8192"):
     else:
         return f"[Groq API Error] {response.text}"
 
+# def call_blackbox_agent(prompt):
+#     url = "https://api.code.blackbox.ai/v1/chat/completions"
+#     headers = {
+#         "Content-Type": "application/json",
+#         "Authorization": f"Bearer {BLACKBOX_API_KEY}"
+#     }
+#     data = {
+#         "message": prompt
+#     }
+#     response = requests.post(url, headers=headers, json=data)
+#     if response.status_code == 200:
+#         return response.json().get("response", response.json())
+#     #else:
+#        # return call_groq_api(prompt)
 def call_blackbox_agent(prompt):
     url = "https://api.code.blackbox.ai/v1/chat/completions"
     headers = {
@@ -37,13 +51,18 @@ def call_blackbox_agent(prompt):
         "Authorization": f"Bearer {BLACKBOX_API_KEY}"
     }
     data = {
-        "message": prompt
+        "model": "code-chat",
+        "messages": [
+            {"role": "system", "content": "You are a helpful coding assistant."},
+            {"role": "user", "content": prompt}
+        ]
     }
     response = requests.post(url, headers=headers, json=data)
     if response.status_code == 200:
-        return response.json().get("response", response.json())
-    #else:
-       # return call_groq_api(prompt)
+        return response.json()["choices"][0]["message"]["content"]
+    # else:
+    #     return call_groq_api(prompt)
+
 
 # --- UTILS ---
 def code_matches_language(code, language):