diff --git a/agent_jarvis/TestPage/css/main.css b/agent_jarvis/TestPage/css/main.css
index 9416efc..4ef1ae3 100644
--- a/agent_jarvis/TestPage/css/main.css
+++ b/agent_jarvis/TestPage/css/main.css
@@ -121,3 +121,10 @@ body {
 .hidden {
   display: none;
 }
+
+.replied-img {
+  width: auto;
+  height: auto;
+  max-height: 400px;
+  max-width: 100%;
+}
\ No newline at end of file
diff --git a/agent_jarvis/TestPage/js/main.js b/agent_jarvis/TestPage/js/main.js
index ad94651..c4b091d 100644
--- a/agent_jarvis/TestPage/js/main.js
+++ b/agent_jarvis/TestPage/js/main.js
@@ -106,7 +106,7 @@
         break;
       case "image": {
         var item = document.createElement("li");
-        item.innerHTML = `<img src="data:image/png;base64,${msgContent}" alt="Image" width="150" height="150">`;
+        item.innerHTML = `<img src="data:image/png;base64,${msgContent}" alt="Image" class="replied-img">`;
         messages.appendChild(item);
         scrollElementToEnd(messages);
         break;