ll;m edit

Author: 0thernet

next/README.md

@@ -3,6 +3,7 @@
 ## now
 
 - real cloning
+- handle different romanization prompts for languages not just chinese
 - summarize kaikki definitions w llm
 
 ## future

next/app/api/translate/route.ts

@@ -16,10 +16,12 @@ if (!process.env.DEEPINFRA_API_KEY) {
 const romanizationSchema = z.object({
   translation: z.string(),
   romanization: z.string(),
+  edited: z.string(),
 });
 
 const standardSchema = z.object({
   translation: z.string(),
+  edited: z.string(),
 });
 
 const inputSchema = z.object({
@@ -34,17 +36,34 @@ export async function POST(request: NextRequest) {
 
     const prompt =
       language === Language.CHINESE_CN
-        ? `Translate the following text into Simplified Chinese (Mandarin).
-Make sure to:
-1. Use only Simplified Chinese characters (not Traditional).
-2. Use pinyin with diacritic marks (not numeric).
-3. Return a valid JSON object with:
-   - "translation" key: Simplified Chinese character translation
-   - "romanization" key: pinyin romanization with diacritic marks (NOT numeric) and syllables separated by spaces.
+        ? `Given the following text, return:
+1. A grammatically corrected version of the original text with proper punctuation, ensuring complete sentences and avoiding fragments
+2. A translation into Simplified Chinese (Mandarin)
+3. Pinyin romanization with diacritic marks
+
+When correcting the text:
+- Join sentence fragments into complete thoughts
+- Ensure proper subject-verb relationships
+- Maintain the original meaning while making the text more cohesive
+
+Return a valid JSON object with:
+- "edited" key: corrected original text with complete sentences
+- "translation" key: Simplified Chinese character translation
+- "romanization" key: pinyin with diacritic marks (NOT numeric) and syllables separated by spaces
 
 Text: ${text}`
-        : `Translate the following text into ${language}. 
-Return a valid JSON object with a single key "translation" whose value is the translation.
+        : `Given the following text, return:
+1. A grammatically corrected version of the original text with proper punctuation, ensuring complete sentences and avoiding fragments
+2. A translation into ${language}
+
+When correcting the text:
+- Join sentence fragments into complete thoughts
+- Ensure proper subject-verb relationships
+- Maintain the original meaning while making the text more cohesive
+
+Return a valid JSON object with:
+- "edited" key: corrected original text with complete sentences
+- "translation" key: translation
 
 Text: ${text}`;
 

next/components/assembly-view.tsx

@@ -203,6 +203,7 @@ const AssemblyView: () => JSX.Element = () => {
                     ...f,
                     translated: result.data.translation,
                     romanization: result.data.romanization,
+                    edited: result.data.edited,
                   }
                 : f
             )

next/components/fragment.tsx

@@ -26,6 +26,7 @@ export type Fragment = {
   createdAt: number;
   translated?: string;
   romanization?: string;
+  edited?: string;
   isPartial?: boolean;
 };
 
@@ -95,11 +96,20 @@ export const FragmentComponent = ({
   if (!showTranslation) {
     return (
       <div
-        className={`border border-black p-4 rounded-md ${
+        className={`border-2 ${
+          fragment.edited && fragment.edited !== fragment.text
+            ? "border-dotted border-black/70 bg-amber-50/30"
+            : "border-solid border-black"
+        } p-4 rounded-md ${
           fragment.type === "speech" ? "bg-black/5" : ""
         } ${isPartial ? "opacity-50" : ""}`}
       >
-        <div>{fragment.text}</div>
+        <div>{fragment.edited || fragment.text}</div>
+        {fragment.edited && fragment.edited !== fragment.text && (
+          <div className="text-xs mt-1 text-gray-500">
+            Original: {fragment.text}
+          </div>
+        )}
       </div>
     );
   }
@@ -208,11 +218,15 @@ export const FragmentComponent = ({
           <ResizablePanel defaultSize={70}>
             <div className="grid grid-cols-2 gap-4 items-start">
               <div
-                className={`border border-black p-4 rounded-md ${
+                className={`border-2 ${
+                  fragment.edited && fragment.edited !== fragment.text
+                    ? "border-dotted border-black/70 bg-amber-50/30"
+                    : "border-solid border-black"
+                } p-4 rounded-md ${
                   fragment.type === "speech" ? "bg-black/5" : ""
                 } ${isPartial ? "opacity-50" : ""}`}
               >
-                {fragment.text}
+                {fragment.edited || fragment.text}
               </div>
               <div className="border border-black p-4 rounded-md">
                 {fragment.translated ? (
@@ -244,11 +258,15 @@ export const FragmentComponent = ({
         <>
           <div className="grid grid-cols-2 gap-4 items-start">
             <div
-              className={`border border-black p-4 rounded-md ${
+              className={`border-2 ${
+                fragment.edited && fragment.edited !== fragment.text
+                  ? "border-dotted border-black/70 bg-amber-50/30"
+                  : "border-solid border-black"
+              } p-4 rounded-md ${
                 fragment.type === "speech" ? "bg-black/5" : ""
               } ${isPartial ? "opacity-50" : ""}`}
             >
-              {fragment.text}
+              {fragment.edited || fragment.text}
             </div>
             <div className="border border-black p-4 rounded-md">
               {fragment.translated ? (