Update app.py

app.py

@@ -30,20 +30,43 @@ 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 = {
+#          "message": prompt
+#      }
+#      response = requests.post(url, headers=headers, json=data)
+#      if response.status_code == 200:
+#         return response.json().get("response", response.json())
+#      else:
+#         return f"[Blackbox API Error] {response.text}"
+
+def call_blackbox_agent(messages, model="gpt-4o"):
+    """
+    messages: list of dicts, e.g.
+      [
+        {"role": "system", "content": "You are a helpful coding assistant."},
+        {"role": "user", "content": "Refactor this code: ..."}
+      ]
+    """
+    url = "http://localhost:4000/v1/chat/completions"
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {BLACKBOX_API_KEY}"
+    }
+    data = {
+        "model": model,
+        "messages": messages
+    }
+    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(messages[-1]["content"])