feat: adjust interface for report chatting

jst-seminar-rostlab-tum · Dec 22, 2024 · 37c55e6 · 37c55e6
1 parent d25365b
commit 37c55e6
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 79 deletions.
diff --git a/src/components/Chatbot/Chatbot.tsx b/src/components/Chatbot/Chatbot.tsx
@@ -96,16 +96,21 @@ export default function HungerMapChatbot() {
     const previousMessages = chats[currentChatIndex].messages;
     let aiResponse = '';
     try {
-      if (chats[currentChatIndex].isReportStarter && chats[currentChatIndex].context) {
+      if (chats[currentChatIndex].isReportStarter && chats[currentChatIndex].reports_country_name) {
         aiResponse = (
           await chatbot.sendMessage(text, {
             previous_messages: previousMessages,
-            context: chats[currentChatIndex].context,
+            reports_country_name: chats[currentChatIndex].reports_country_name,
           })
         ).response;
         chats[currentChatIndex].isReportStarter = false;
       } else {
-        aiResponse = (await chatbot.sendMessage(text, { previous_messages: previousMessages })).response;
+        aiResponse = (
+          await chatbot.sendMessage(text, {
+            previous_messages: previousMessages,
+            reports_country_name: chats[currentChatIndex].reports_country_name,
+          })
+        ).response;
       }
     } catch (err) {
       if (err instanceof APIError) {

diff --git a/src/domain/contexts/ChatbotContext.tsx b/src/domain/contexts/ChatbotContext.tsx
@@ -83,9 +83,8 @@ export function ChatbotProvider({ children }: { children: ReactNode }) {
    * Initiates a chat with a report based on the provided country name and report content.
    * If a chat for that specific already exists, it opens that chat; otherwise, it creates a new one.
    * @param countryName - The name of the country related to the report.
-   * @param reportURL - The URL of the report to be processed.
    */
-  const initiateChatAboutReport = async (countryName: string, reportURL: string) => {
+  const initiateChatAboutReport = async (countryName: string) => {
     const reportChatIndex = chats.findIndex((chat) => chat.title === `Report ${countryName}`);
     const reportChatExists = reportChatIndex !== -1;
 
@@ -94,26 +93,16 @@ export function ChatbotProvider({ children }: { children: ReactNode }) {
       return;
     }
 
-    const { extractPdfText } = await import('@/operations/chatbot/ExtractPdfText');
-    let reportText;
-    try {
-      reportText = await extractPdfText(reportURL, showSnackBar);
-    } catch {
-      reportText = '';
-    }
-
     const assistantMessage = {
       id: crypto.randomUUID(),
-      content: reportText
-        ? `Hey, how can I help you with this report about ${countryName}?`
-        : `Hey, unfortunately I'm currently unable to answer questions about this report. You can try it later or chat with me about other things!`,
+      content: `Hey, how can I help you with this report about ${countryName}?`,
       role: SenderRole.ASSISTANT,
     };
 
     const newChat: IChat = {
       id: chats.length + 1,
       title: `Report ${countryName}`,
-      context: reportText,
+      reports_country_name: countryName,
       isReportStarter: true,
       messages: [assistantMessage],
       isTyping: false,

diff --git a/src/domain/entities/chatbot/BackendCommunication.ts b/src/domain/entities/chatbot/BackendCommunication.ts
@@ -1,7 +1,7 @@
 import { IMessage } from './Chatbot';
 
 export interface QueryRequest {
-  context?: string;
+  reports_country_name?: string;
   query: string;
   version?: number;
   chatbot_type?: string;

diff --git a/src/domain/entities/chatbot/Chatbot.ts b/src/domain/entities/chatbot/Chatbot.ts
@@ -10,7 +10,7 @@ export interface IMessage {
 export interface IChat {
   id: number;
   title: string;
-  context?: string;
+  reports_country_name?: string;
   isReportStarter?: boolean;
   messages: IMessage[];
   isTyping: boolean;

diff --git a/src/infrastructure/repositories/ChatbotRepositoryImpl.ts b/src/infrastructure/repositories/ChatbotRepositoryImpl.ts
@@ -14,7 +14,7 @@ export default class ChatbotRepositoryImpl implements ChatbotRepository {
   async sendMessage(message: string, options: Partial<Omit<QueryRequest, 'query'>>): Promise<QueryResponse> {
     try {
       const payload: QueryRequest = {
-        context: options.context || '',
+        reports_country_name: options.reports_country_name || '',
         query: message,
         version: options.version || 1,
         chatbot_type: options.chatbot_type || 'gpt-4',

diff --git a/src/operations/chatbot/ExtractPdfText.ts b/src/operations/chatbot/ExtractPdfText.ts