Merge pull request #190 from CS3219-AY2425S1/feat/collab-service/ai-chat

Add AI Chat

Author: dylkaw

.env.sample

@@ -25,6 +25,7 @@ MATCHING_SVC_PORT=6969
 ## Collab service variables
 COLLAB_SVC_PORT=3002
 COLLAB_SVC_DB_URI=
+OPENAI_API_KEY=
 
 ## Redis variables
 REDIS_PORT=6379

collab-service/app/controller/ai-controller.js

@@ -0,0 +1,19 @@
+import { sendAiMessage } from "../model/repository.js";
+
+// send ai message
+export async function sendAiMessageController(req, res) {
+  const { message } = req.body;
+  if (!message) {
+    return res.status(400).json({ error: "Message content is required" });
+  }
+
+  const data = await sendAiMessage(message);
+  const aiResponse =
+    data.choices?.[0]?.message?.content || "No response from AI";
+
+  if (aiResponse) {
+    res.status(200).json({ data });
+  } else {
+    res.status(500).json({ error: "Failed to retrieve AI response" });
+  }
+}

collab-service/app/model/repository.js

@@ -120,3 +120,23 @@ export async function getQuestionIdByRoomId(roomId) {
     return null;
   }
 }
+
+export async function sendAiMessage(message) {
+  try {
+    const response = await fetch("https://api.openai.com/v1/chat/completions", {
+      method: "POST",
+      headers: {
+        'Content-Type': "application/json",
+        'Authorization': `Bearer ${process.env.OPENAI_API_KEY}`,
+      },
+      body: JSON.stringify({
+        model: "gpt-3.5-turbo",
+        messages: [{ role: "user", content: message }],
+      }),
+    });
+    const data = await response.json();
+    return data;
+  } catch (error) {
+    console.error("Error in sending AI message:", error);
+  }
+}

collab-service/app/routes/collab-routes.js

@@ -7,6 +7,7 @@ import {
   getRoomChatHistory,
   getQuestionId,
 } from "../controller/collab-controller.js";
+import { sendAiMessageController } from "../controller/ai-controller.js";
 
 const router = express.Router();
 
@@ -22,4 +23,6 @@ router.get("/chat-history/:roomId", getRoomChatHistory);
 
 router.get("/rooms/:roomId/questionId", getQuestionId);
 
+router.post("/send-ai-message", sendAiMessageController)
+
 export default router;

docker-compose.yml

@@ -79,4 +79,5 @@ services:
       - $COLLAB_SVC_PORT:$COLLAB_SVC_PORT
     environment:
       - PORT=$COLLAB_SVC_PORT
-      - DB_URI=$COLLAB_SVC_DB_URI
+      - DB_URI=$COLLAB_SVC_DB_URI
+      - OPENAI_API_KEY=$OPENAI_API_KEY

frontend/Dockerfile

@@ -13,8 +13,8 @@ ENV NEXT_PUBLIC_BASE_URI=$BASE_URI \
     NEXT_PUBLIC_USER_SVC_PORT=$USER_SVC_PORT \
     NEXT_PUBLIC_QUESTION_SVC_PORT=$QUESTION_SVC_PORT \
     NEXT_PUBLIC_MATCHING_SVC_PORT=$MATCHING_SVC_PORT  \
-    NEXT_PUBLIC_COLLAB_SVC_PORT=$COLLAB_SVC_PORT
-
+    NEXT_PUBLIC_COLLAB_SVC_PORT=$COLLAB_SVC_PORT 
+    
 # Production build stage
 FROM base AS build
 COPY . .

frontend/components/collab/chat.tsx

@@ -10,6 +10,7 @@ import { Send } from "lucide-react";
 import { io, Socket } from "socket.io-client";
 import { useAuth } from "@/app/auth/auth-context";
 import LoadingScreen from "@/components/common/loading-screen";
+import { sendAiMessage } from "@/lib/api/openai/send-ai-message";
 import { getChatHistory } from "@/lib/api/collab-service/get-chat-history";
 
 interface Message {
@@ -134,7 +135,7 @@ export default function Chat({ roomId }: { roomId: string }) {
     scrollWithDelay();
   }, [partnerMessages, aiMessages, chatTarget]);
 
-  const sendMessage = () => {
+  const sendMessage = async () => {
     if (!newMessage.trim() || !socket || !isConnected || !own_user_id) return;
 
     if (chatTarget === "partner") {
@@ -151,6 +152,18 @@ export default function Chat({ roomId }: { roomId: string }) {
         timestamp: new Date(),
       };
       setAiMessages((prev) => [...prev, message]);
+      const response = await sendAiMessage(newMessage);
+      const data = await response.json();
+      const aiMessage = {
+        id: crypto.randomUUID(),
+        userId: "ai",
+        text:
+          data.data.choices && data.data.choices[0]?.message?.content
+            ? data.data.choices[0].message.content
+            : "An error occurred. Please try again.",
+        timestamp: new Date(),
+      };
+      setAiMessages((prev) => [...prev, aiMessage]);
     }
 
     setNewMessage("");

frontend/lib/api/openai/send-ai-message.ts

@@ -0,0 +1,17 @@
+import { collabServiceUri } from "@/lib/api/api-uri";
+
+export const sendAiMessage = async (message: string) => {
+  console.log(`{collabServiceUri(window.location.hostname)}/send-ai-message`);
+  console.log(message);
+  const response = await fetch(
+    `${collabServiceUri(window.location.hostname)}/collab/send-ai-message`,
+    {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      body: JSON.stringify({ message: message }),
+    }
+  );
+  return response;
+};