✨ feat(frontend): Display predicted spam label in post

Author: LauraRandl

chart/templates/microblog-service.yaml

@@ -68,6 +68,8 @@ spec:
               value: {{ quote .Values.microblogService.deployment.container.env.USER_AUTH_SERVICE_ADDRESS }}
             - name: RAG_SERVICE_ADDRESS
               value: {{ quote .Values.microblogService.deployment.container.env.RAG_SERVICE_ADDRESS }}
+            - name: RAG_SERVICE_PORT
+              value: {{ quote .Values.microblogService.deployment.container.env.RAG_SERVICE_PORT }}
             - name: OPENTRACING_JAEGER_ENABLED
               value: {{ quote .Values.microblogService.deployment.container.env.OPENTRACING_JAEGER_ENABLED }}
             - name: JAEGER_SERVICE_NAME

chart/values.yaml

@@ -279,7 +279,8 @@ microblogService:
         OPENTRACING_JAEGER_ENABLED: false
         REDIS_SERVICE_ADDRESS: unguard-redis
         USER_AUTH_SERVICE_ADDRESS: unguard-user-auth-service
-        RAG_SERVICE_ADDRESS: unguard-rag-service:8000
+        RAG_SERVICE_ADDRESS: unguard-rag-service
+        RAG_SERVICE_PORT: 8000
 
 # Status Service
 statusService:

src/frontend-nextjs/app/post/page.tsx

@@ -33,6 +33,7 @@ function SinglePost() {
                     postId={postData.postId}
                     timestamp={postData.timestamp}
                     username={postData.username}
+                    isSpamPredictedLabel={postData.isSpamPredictedLabel}
                 />
             )}
         </div>

src/frontend-nextjs/components/Timeline/Post.tsx

@@ -10,13 +10,15 @@ import { BASE_PATH } from '@/constants';
 import { useCheckLogin } from '@/hooks/queries/useCheckLogin';
 import { LikeButton } from '@/components/Timeline/LikeButton';
 import { ErrorCard } from '@/components/ErrorCard';
+import { PostSpamPrediction } from '@/components/Timeline/PostSpamPrediction';
 
 export interface PostProps {
     username: string;
     timestamp: string;
     body: string;
     imageUrl?: string;
     postId: string;
+    isSpamPredictedLabel?: boolean;
 }
 
 export function Post(props: PostProps) {
@@ -69,7 +71,11 @@ export function Post(props: PostProps) {
                     )}
                     <p>{props.body}</p>
                 </CardBody>
-                <CardFooter className='gap-3 justify-end px-3'>
+                <CardFooter className='gap-3 justify-between px-3'>
+                    <div className='flex items-center'>
+                        <PostSpamPrediction isSpamPredictedLabel={props.isSpamPredictedLabel} />
+                    </div>
+
                     {isLoggedIn && (
                         <div className='flex gap-1'>
                             <ErrorBoundary fallbackRender={(props) => <ErrorCard message={props.error.message} />}>

src/frontend-nextjs/components/Timeline/PostSpamPrediction.tsx

@@ -0,0 +1,34 @@
+import React from 'react';
+import {Alert} from "@heroui/react";
+
+export interface PostSpamPredictionProps {
+    isSpamPredictedLabel?: boolean | null;
+}
+
+
+
+export function PostSpamPrediction(props: Readonly<PostSpamPredictionProps>) {
+    console.log("PostSpamPrediction props:", props);
+    if (props.isSpamPredictedLabel == null) return null;
+
+
+    if (props.isSpamPredictedLabel) {
+        const color='danger'
+        return (
+            <div key={color} className='w-full flex items-center my-3'>
+                <div>
+                    <Alert color={color} title={'Potential Spam Detected'} />
+                </div>
+            </div>
+        );
+    }
+
+    const color = 'primary'
+    return (
+        <div key={color} className='w-full flex items-center my-3'>
+            <div>
+                <Alert color={color} title={'No Spam Detected'} />
+            </div>
+        </div>
+    );
+}

src/frontend-nextjs/components/Timeline/Timeline.tsx

@@ -1,8 +1,7 @@
 'use client';
 import { Card, Spacer, Spinner } from '@heroui/react';
 
-import { Post } from '@/components/Timeline/Post';
-import { PostProps } from '@/components/Timeline/Post';
+import { Post, PostProps } from '@/components/Timeline/Post';
 
 interface TimelineProps {
     posts: PostProps[] | undefined;
@@ -31,6 +30,7 @@ export function Timeline({ posts, isLoading }: TimelineProps) {
                         postId={post.postId}
                         timestamp={post.timestamp}
                         username={post.username}
+                        isSpamPredictedLabel={post.isSpamPredictedLabel}
                     />
                     <Spacer y={4} />
                 </div>

src/microblog-service/src/main/java/org/dynatrace/microblog/MicroblogController.java

@@ -23,11 +23,7 @@
 import org.dynatrace.microblog.dto.PostId;
 import org.dynatrace.microblog.dto.SerializedPost;
 import org.dynatrace.microblog.dto.User;
-import org.dynatrace.microblog.exceptions.FollowYourselfException;
-import org.dynatrace.microblog.exceptions.InvalidJwtException;
-import org.dynatrace.microblog.exceptions.InvalidUserException;
-import org.dynatrace.microblog.exceptions.NotLoggedInException;
-import org.dynatrace.microblog.exceptions.UserNotFoundException;
+import org.dynatrace.microblog.exceptions.*;
 import org.dynatrace.microblog.form.PostForm;
 import org.dynatrace.microblog.ragservice.RAGServiceClient;
 import org.dynatrace.microblog.redis.RedisClient;
@@ -73,6 +69,7 @@ public MicroblogController(Tracer tracer, PostSerializer postSerializer) {
         String redisServiceAddress;
         String userAuthServiceAddress;
         String ragServiceAddress;
+        String ragServicePort;
         if (System.getenv("REDIS_SERVICE_ADDRESS") != null) {
             redisServiceAddress = System.getenv("REDIS_SERVICE_ADDRESS");
             logger.info("REDIS_SERVICE_ADDRESS set to {}", redisServiceAddress);
@@ -93,13 +90,20 @@ public MicroblogController(Tracer tracer, PostSerializer postSerializer) {
             ragServiceAddress = System.getenv("RAG_SERVICE_ADDRESS");
             logger.info("RAG_SERVICE_ADDRESS set to {}", ragServiceAddress);
         } else {
-            ragServiceAddress = "localhost:8000";
-            logger.warn("No RAG_SERVICE_ADDRESS environment variable defined, falling back to localhost:8000.");
+            ragServiceAddress = "localhost";
+            logger.warn("No RAG_SERVICE_ADDRESS environment variable defined, falling back to localhost.");
+        }
+        if (System.getenv("RAG_SERVICE_PORT") != null) {
+            ragServicePort = System.getenv("RAG_SERVICE_PORT");
+            logger.info("RAG_SERVICE_PORT set to {}", ragServicePort);
+        } else {
+            ragServicePort = "8000";
+            logger.warn("No RAG_SERVICE_PORT environment variable defined, falling back to 8000.");
         }
 
         this.userAuthServiceClient = new UserAuthServiceClient(userAuthServiceAddress);
         this.redisClient = new RedisClient(redisServiceAddress, this.userAuthServiceClient, tracer);
-        this.ragServiceClient = new RAGServiceClient(ragServiceAddress);
+        this.ragServiceClient = new RAGServiceClient(ragServiceAddress, ragServicePort);
         this.postSerializer = postSerializer;
     }
 
@@ -193,9 +197,15 @@ public PostId post(@RequestBody PostForm postForm, @CookieValue(value = "jwt", r
         CompletableFuture.runAsync(() -> {
             try {
                 String spamClassificationResult = ragServiceClient.getSpamClassification(postForm.getContent());
-                logger.info("RAG spam classification result for post with ID {}: {}", postId, spamClassificationResult);
-                boolean isSpam = spamClassificationResult.trim().equalsIgnoreCase("spam");
-                redisClient.setSpamPredictedLabel(postId, isSpam);
+                if(spamClassificationResult == null || spamClassificationResult.trim().isEmpty()) {
+                    throw new InvalidSpamPredictionException("RAG spam classification result is null or empty for post with ID " + postId);
+                } else if (spamClassificationResult.trim().equalsIgnoreCase("not_spam")) {
+                    redisClient.setSpamPredictedLabel(postId, false);
+                } else if (spamClassificationResult.trim().equalsIgnoreCase("spam")) {
+                    redisClient.setSpamPredictedLabel(postId, true);
+                } else {
+                    throw new InvalidSpamPredictionException("RAG spam classification result is invalid for post with ID " + postId + ": " + spamClassificationResult);
+                }
             } catch (Exception e) {
                 logger.warn("RAG spam classification failed for post with ID {}", postId, e);
             }

src/microblog-service/src/main/java/org/dynatrace/microblog/exceptions/InvalidSpamPredictionException.java

@@ -0,0 +1,11 @@
+package org.dynatrace.microblog.exceptions;
+
+import org.springframework.http.HttpStatus;
+import org.springframework.web.bind.annotation.ResponseStatus;
+
+@ResponseStatus(code = HttpStatus.BAD_REQUEST, reason = "Invalid spam prediction")
+public class InvalidSpamPredictionException extends Exception {
+  public InvalidSpamPredictionException(String message) {
+    super(message);
+  }
+}

src/microblog-service/src/main/java/org/dynatrace/microblog/ragservice/RAGServiceClient.java

@@ -21,9 +21,11 @@ public class RAGServiceClient {
     private final Logger logger = LoggerFactory.getLogger(org.dynatrace.microblog.ragservice.RAGServiceClient.class);
 
     private final String ragServiceHost;
+    private final int ragServicePort;
 
-    public RAGServiceClient(String ragServiceHost) {
+    public RAGServiceClient(String ragServiceHost, String ragsServicePort) {
         this.ragServiceHost = ragServiceHost;
+        this.ragServicePort = Integer.parseInt(ragsServicePort);
 
         TracingInterceptor tracingInterceptor = new TracingInterceptor(
             GlobalTracer.get(),
@@ -47,10 +49,15 @@ public String getSpamClassification(String postText) throws InvalidJwtException,
         RequestBody body = RequestBody.create(
             MediaType.parse("application/json"), jsonRequest);
 
-        String requestUrl = "http://" + this.ragServiceHost + "/classifyPost";
+        HttpUrl url = new HttpUrl.Builder()
+                .scheme("http")
+                .host(this.ragServiceHost)
+                .port(this.ragServicePort)
+                .addPathSegment("classifyPost")
+                .build();
 
         Request request = new Request.Builder()
-            .url(requestUrl)
+            .url(url)
             .post(body)
             .build();