diff --git a/client/src/components/PromptInput/PromptInput.jsx b/client/src/components/PromptInput/PromptInput.jsx
index 56191fe..0596ab3 100644
--- a/client/src/components/PromptInput/PromptInput.jsx
+++ b/client/src/components/PromptInput/PromptInput.jsx
@@ -10,6 +10,9 @@ export default function PromptInput() {
   const [error, setError] = useState("");
   const dispatch = useDispatch();
 
+  const handleInputChange = (e) => {
+    setInputValue(e.target.value);
+  };
   const handleInputChange = (e) => {
     setInputValue(e.target.value);
   };
@@ -65,6 +68,10 @@ export default function PromptInput() {
     setInputValue("");
     setError("");
   };
+    // Reset the input value and error state
+    setInputValue("");
+    setError("");
+  };
 
   return (
     <div
diff --git a/client/src/pages/QueryPage.jsx b/client/src/pages/QueryPage.jsx
index ace4ccc..24ac261 100644
--- a/client/src/pages/QueryPage.jsx
+++ b/client/src/pages/QueryPage.jsx
@@ -2,23 +2,32 @@ import { Grid } from "@mui/material";
 import CodeEditor from "../components/CodeEditor/CodeEditor";
 import PromptInput from "../components/PromptInput/PromptInput";
 import Output from "../components/Output/Output";
+import { useState } from "react";
+
+const myMap = new Map([
+  ["code", [6, 6]],
+  ["text", [5, 7]],
+]);
 
 export default function QueryPage() {
-	return (
-		<>
-			<h1 style={{ textAlign: "center" }}>Monaco Editor Example</h1>
+  const [type, setType] = useState("text");
+  const [length1, length2] = myMap.get(type);
+
+  return (
+    <>
+      <h1 style={{ textAlign: "center" }}>Monaco Editor Example</h1>
 
-			<div style={{ paddingLeft: "1rem", paddingRight: "1rem" }}>
-				<Grid container spacing={2}>
-					<Grid item xs={6}>
-						<CodeEditor defaultCode={"# Paste/Enter your code here"} />
-						{/* <PromptInput /> */}
-					</Grid>
-					<Grid item xs={6}>
-						<Output />
-					</Grid>
-				</Grid>
-			</div>
-		</>
-	);
+      <div style={{ paddingLeft: "1rem", paddingRight: "1rem" }}>
+        <Grid container spacing={2}>
+          <Grid item xs={length1}>
+            {/* <CodeEditor defaultCode={"# Paste/Enter your code here"} /> */}
+            <PromptInput />
+          </Grid>
+          <Grid item xs={length2}>
+            <Output />
+          </Grid>
+        </Grid>
+      </div>
+    </>
+  );
 }