Update app.py

app.py

@@ -30,38 +30,21 @@ 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 = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {BLACKBOX_API_KEY}"
-    }
-    data = {
-        "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()["choices"][0]["message"]["content"]
-    # else:
-    #     return call_groq_api(prompt)
+ 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)
+
 
 
 # --- UTILS ---