diff --git a/src/api/chat.ts b/src/api/chat.ts
index c497705..4c49517 100644
--- a/src/api/chat.ts
+++ b/src/api/chat.ts
@@ -41,35 +41,31 @@ export const Generate = async ({
       normalizedImage = [''];
     }
 
-    if (normalizedImage.some(img => img !== '') || tool) {
-      // Normal axios request
-      res = await axios.post(`${BASE_URL}/generate/msg`, {
-        email,
-        message,
-        newConvo,
-        conversationId,
-        provider,
-        modelName,
-        image: normalizedImage,
-        tool,
-        files,
-      }, { headers: getAuthHeaders() });
-      return res.data;
-    } else {
-      // Handle the else case (you had an empty else block)
-      res = await axios.post(`${BASE_URL}/generate/msg`, {
-        email,
-        message,
-        newConvo,
-        conversationId,
-        provider,
-        modelName,
-        image: normalizedImage,
-        tool,
-        files,
-      }, { headers: getAuthHeaders() });
-      return res.data;
-    }
+    const hasMediaOrTool = normalizedImage.some(img => img !== '') || tool;
+
+const payload: any = {
+  email,
+  message,
+  newConvo,
+  conversationId,
+  provider,
+  modelName,
+  files,
+};
+
+if (hasMediaOrTool) {
+  payload.image = normalizedImage;
+  payload.tool = tool;
+}
+
+res = await axios.post(
+  `${BASE_URL}/generate/msg`,
+  payload,
+  { headers: getAuthHeaders() }
+);
+
+return res.data;
+
   } catch (err: any) {
     const message =
       err.response?.data || "Response Generation Failed. Try again later.";