Update app.js

app.js

@@ -10,7 +10,7 @@ const app = express();
 const PORT = process.env.PORT || 7860;
 
 app.use(cors());
-app.use(express.json({ limit: '50mb' })); // Increased limit for images
+app.use(express.json({ limit: '50mb' }));
 
 // --- SYSTEM PROMPT DEFINITIONS ---
 const CLAUDE_SYSTEM_PROMPT = "You are a pro. Provide elite, high-level technical responses.";
@@ -30,10 +30,9 @@ const azureOpenAI = new OpenAI({
     defaultHeaders: { "api-key": "7U3m9NRkE38ThSWTr92hMgQ4hDCUFI9MAnFNrCgRL7MhdvckfTXwJQQJ99CBACHYHv6XJ3w3AAAAACOGV22P" }
 });
 
+// Standard Generation
 app.post('/api/generate', async (req, res) => {
     const { model, prompt, system_prompt} = req.body;
-    console.log(`[TRAFFIC] Request for ${model}`);
-
     try {
         if (model === "claude") {
             const command = new ConverseCommand({
@@ -46,7 +45,6 @@ app.post('/api/generate', async (req, res) => {
                     output_config: { effort: "high" }
                 }
             });
-            
             const response = await bedrockClient.send(command);
             const text = response.output.message.content.find(b => b.text)?.text;
             res.json({ success: true, data: text });
@@ -62,15 +60,15 @@ app.post('/api/generate', async (req, res) => {
             res.json({ success: true, data: response.choices[0].message.content });
         }
     } catch (err) {
-        console.error(`❌ [${model.toUpperCase()} ERROR]:`, err.name, err.message);
-        res.status(500).json({ success: false, error: `${err.name}: ${err.message}` });
+        console.error(`❌ [${model.toUpperCase()} ERROR]:`, err.message);
+        res.status(500).json({ success: false, error: err.message });
     }
 });
 
-// --- STREAMING ENDPOINT WITH IMAGE SUPPORT ---
+// Streaming Generation
 app.post('/api/stream', async (req, res) => {
-    const { model, prompt, system_prompt, images } = req.body; // Expect images array (base64)
-    console.log(`[STREAM] Request for ${model} ${images?.length ? 'with images' : ''}`);
+    const { model, prompt, system_prompt, images } = req.body;
+    console.log(`[STREAM] Request for ${model}`);
 
     res.setHeader('Content-Type', 'text/plain; charset=utf-8');
     res.setHeader('Transfer-Encoding', 'chunked');
@@ -79,20 +77,11 @@ app.post('/api/stream', async (req, res) => {
 
     try {
         if (model === "claude") {
-            // Construct Content Block for Claude
             let contentBlock = [{ text: prompt }];
-            
             if (images && images.length > 0) {
-                // Prepend images to content block
-                // Bedrock expects pure base64 without data prefix
                 const imageBlocks = images.map(imgStr => {
                     const base64Data = imgStr.replace(/^data:image\/\w+;base64,/, "");
-                    return {
-                        image: {
-                            format: 'png', // Assuming PNG or generic byte handling
-                            source: { bytes: Buffer.from(base64Data, 'base64') }
-                        }
-                    };
+                    return { image: { format: 'png', source: { bytes: Buffer.from(base64Data, 'base64') } } };
                 });
                 contentBlock = [...imageBlocks, ...contentBlock];
             }
@@ -102,42 +91,23 @@ app.post('/api/stream', async (req, res) => {
                 system: [{ text: system_prompt || CLAUDE_SYSTEM_PROMPT }],
                 messages: [{ role: "user", content: contentBlock }],
                 inferenceConfig: { maxTokens: 48000, temperature: 1 },
-                additionalModelRequestFields: { 
-                    thinking: { type: "adaptive" },
-                    output_config: { effort: "high" }
-                }
+                additionalModelRequestFields: { thinking: { type: "adaptive" }, output_config: { effort: "high" } }
             });
 
             const response = await bedrockClient.send(command);
-            
             for await (const chunk of response.stream) {
                 if (chunk.contentBlockDelta) {
                     const delta = chunk.contentBlockDelta.delta;
-                    if (delta.reasoningContent && delta.reasoningContent.text) {
-                        res.write(`__THINK__${delta.reasoningContent.text}`);
-                    } else if (delta.text) {
-                        res.write(delta.text);
-                    }
+                    if (delta.reasoningContent?.text) res.write(`__THINK__${delta.reasoningContent.text}`);
+                    else if (delta.text) res.write(delta.text);
                 }
             }
             res.end();
-
         } else {
-            // Construct Content Block for OpenAI
-            let messagesPayload = [
-                { role: "system", content: system_prompt || GPT_SYSTEM_PROMPT }
-            ];
-
-            let userContent = [];
+            let messagesPayload = [{ role: "system", content: system_prompt || GPT_SYSTEM_PROMPT }];
             if (images && images.length > 0) {
-                // OpenAI supports mixed content array
-                userContent.push({ type: "text", text: prompt });
-                images.forEach(imgStr => {
-                    userContent.push({
-                        type: "image_url",
-                        image_url: { url: imgStr } // OpenAI accepts data URI directly
-                    });
-                });
+                let userContent = [{ type: "text", text: prompt }];
+                images.forEach(img => userContent.push({ type: "image_url", image_url: { url: img } }));
                 messagesPayload.push({ role: "user", content: userContent });
             } else {
                 messagesPayload.push({ role: "user", content: prompt });
@@ -152,11 +122,8 @@ app.post('/api/stream', async (req, res) => {
 
             for await (const chunk of stream) {
                 const delta = chunk.choices[0]?.delta;
-                if (delta?.reasoning_content) {
-                    res.write(`__THINK__${delta.reasoning_content}`);
-                } else if (delta?.content) {
-                    res.write(delta.content);
-                }
+                if (delta?.reasoning_content) res.write(`__THINK__${delta.reasoning_content}`);
+                else if (delta?.content) res.write(delta.content);
             }
             res.end();
         }
@@ -167,8 +134,4 @@ app.post('/api/stream', async (req, res) => {
     }
 });
 
-app.post('/', async (req, res) => {
-res.json({ success: true, data: response.choices[0].message.content });
-});
-
-app.listen(PORT, '0.0.0.0', () => console.log(`Main AI Agent live on port ${PORT}`));
+app.listen(PORT, '0.0.0.0', () => console.log(`Main AI Agents live on port ${PORT}`));