commit 014

src/App.js

@@ -9,14 +9,25 @@ export default function App() {
   const [messages, setMessages] = useState([]);
   const [lastCode, setLastCode] = useState("");
   const [loading, setLoading] = useState(false);
+
+  const detectLanguage = (filename) => {
+  const ext = filename.split('.').pop();
+  switch (ext) {
+    case "js": return "javascript";
+    case "py": return "python";
+    case "java": return "java";
+    case "cpp": return "cpp";
+    case "ts": return "typescript";
+    case "cs": return "csharp";
+    default: return "";
+  }
+};
   
   const handleSend = async (msg) => {
     const newHistory = [...messages, { role: "user", content: msg }];
     setMessages(newHistory);
     setLoading(true);
 
-   // let botMsg = { role: "assistant", content: "" };
-    //setMessages((prev) => [...prev, botMsg]);
 
     if (msg.includes("```")) setLastCode(msg);
 
@@ -27,7 +38,6 @@ export default function App() {
       history: messages,
       onChunk: (chunk) => {
         fullContent += chunk;
-       // setMessages((prev) => [...prev.slice(0, -1), { ...botMsg }]);
        setMessages([...newHistory, { role: "assistant", content: fullContent }]);
       },
       onDone: () => {
@@ -54,14 +64,22 @@ export default function App() {
       <div className="main">
         <ChatBox messages={messages} loading={loading}/>
         <div className="buttons">
-          <FileUpload onCodeParsed={(code) => {
-            const formatted = "```javascript\n" + code + "\n```";
-            setLastCode(formatted);
-            handleSend("Here's my code:\n\n" + formatted);
-          }} />
+          <FileUpload onCodeParsed={(code, filename) => {
+    const lang = detectLanguage(filename);
+    const formatted = `\`\`\`${lang}\n${code}\n\`\`\``;
+    setLastCode(formatted);
+    setMessages((prev) => [
+      ...prev,
+      { role: "user", content: `Here's my code:\n\n${formatted}` },
+    ]);
+  }} />
           <button onClick={handleExplain}>Explain Code</button>
           <button onClick={handleFixBug}>Fix Bug</button>
           {/* <button onClick={handleStackOverflow}>StackOverflow</button> */}
+          <button onClick={() => {
+  setMessages([]);
+  setLastCode("");
+}}>Clear Chat</button>
         </div>
         <ChatInput onSend={handleSend} />
       </div>

src/components/CodeBlock.jsx

@@ -1,9 +1,20 @@
 import { Prism as SyntaxHighlighter } from "react-syntax-highlighter";
+import { oneDark } from "react-syntax-highlighter/dist/esm/styles/prism";
 import { vscDarkPlus } from "react-syntax-highlighter/dist/esm/styles/prism";
 
 export default function CodeBlock({ code }) {
   return (
-    <SyntaxHighlighter language="javascript" style={vscDarkPlus}>
+    <SyntaxHighlighter
+      language="javascript" // fallback default
+      style={oneDark}
+      showLineNumbers
+      wrapLines
+      customStyle={{
+        fontSize: "0.9rem",
+        borderRadius: "8px",
+        padding: "1em",
+      }}
+    >
       {code}
     </SyntaxHighlighter>
   );