initial ask edit toggle

Author: STetsing

libs/remix-ai-core/src/agents/workspaceAgent.ts

@@ -93,16 +93,51 @@ export class workspaceAgent {
       if (typeof payload === 'string') {
         payload = JSON.parse(payload)
       }
-      let modifiedFilesMarkdown = '### List of Modified Files\n'
+      let modifiedFilesMarkdown = '#### **List of Modified Files**\n'
+      let createdFilesMarkdown = '#### **List of Created Files**\n'
+      let hasCreatedFiles = false
+
       for (const file of payload.files) {
         if (!Object.values(SupportedFileExtensions).some(ext => file.fileName.endsWith(ext))) continue;
-        // const fileContent = await this.plugin.call('codeFormatter', 'format', file.fileName, file.content, true);
-        await statusCallback?.(`Showing diff for ${file.fileName}...`)
-        await this.plugin.call('editor', 'showCustomDiff', file.fileName, file.content)
-        modifiedFilesMarkdown += `- ${file.fileName}\n`
+
+        try {
+          const fileExists = await this.plugin.call('fileManager', 'exists', file.fileName)
+
+          if (fileExists) {
+            await this.plugin.call('editor', 'showCustomDiff', file.fileName, file.content)
+            modifiedFilesMarkdown += `- ${file.fileName}\n`
+          } else {
+            await statusCallback?.(`Creating file ${file.fileName}...`)
+
+            const dirPath = file.fileName.substring(0, file.fileName.lastIndexOf('/'))
+            if (dirPath && dirPath.length > 0) {
+              try {
+                await this.plugin.call('fileManager', 'mkdir', dirPath)
+              } catch (mkdirError) {
+                // Directory already exist, just continue
+              }
+            }
+
+            await this.plugin.call('fileManager', 'writeFile', file.fileName, "")
+            await new Promise(resolve => setTimeout(resolve, 2000)) // wait so the content is written before diffing
+            await this.plugin.call('editor', 'showCustomDiff', file.fileName, file.content)
+            createdFilesMarkdown += `- ${file.fileName}\n`
+            hasCreatedFiles = true
+          }
+        } catch (fileError) {
+          console.warn(`Error processing file ${file.fileName}:`, fileError)
+        }
       }
+
       await statusCallback?.('Workspace modifications complete!')
-      return modifiedFilesMarkdown
+
+      // Build result markdown
+      let result = modifiedFilesMarkdown
+      if (hasCreatedFiles) {
+        result += '\n' + createdFilesMarkdown
+      }
+
+      return result || 'No files modified'
     } catch (error) {
       console.warn('Error writing generation results:', error);
       return 'No files modified'

libs/remix-ui/remix-ai-assistant/src/components/prompt.tsx

@@ -38,6 +38,8 @@ export interface PromptAreaProps {
   aiAssistantGroupList: groupListType[]
   textareaRef?: React.RefObject<HTMLTextAreaElement>
   maximizePanel: () => Promise<void>
+  aiMode: 'ask' | 'edit'
+  setAiMode: React.Dispatch<React.SetStateAction<'ask' | 'edit'>>
 }
 
 const _paq = (window._paq = window._paq || [])
@@ -73,7 +75,9 @@ export const PromptArea: React.FC<PromptAreaProps> = ({
   aiContextGroupList,
   aiAssistantGroupList,
   textareaRef,
-  maximizePanel
+  maximizePanel,
+  aiMode,
+  setAiMode
 }) => {
 
   return (
@@ -109,7 +113,32 @@ export const PromptArea: React.FC<PromptAreaProps> = ({
             {contextChoice === 'current' && <span data-id="aiContext-current">{'Current File'}</span>}
           </button>
 
-          <div className="d-flex justify-content-center align-items-center">
+          <div className="d-flex justify-content-center align-items-center gap-2">
+            {/* Ask/Edit Mode Toggle */}
+            <div className="btn-group btn-group-sm" role="group">
+              <button
+                type="button"
+                className={`btn ${aiMode === 'ask' ? 'btn-primary' : 'btn-outline-secondary'} px-2`}
+                onClick={() => {
+                  setAiMode('ask')
+                  _paq.push(['trackEvent', 'remixAI', 'ModeSwitch', 'ask'])
+                }}
+                title="Ask mode - Chat with AI"
+              >
+                Ask
+              </button>
+              <button
+                type="button"
+                className={`btn ${aiMode === 'edit' ? 'btn-primary' : 'btn-outline-secondary'} px-2`}
+                onClick={() => {
+                  setAiMode('edit')
+                  _paq.push(['trackEvent', 'remixAI', 'ModeSwitch', 'edit'])
+                }}
+                title="Edit mode - Edit workspace code"
+              >
+                Edit
+              </button>
+            </div>
             <CustomTooltip
               tooltipText={<TooltipContent />}
               delay={{ show: 1000, hide: 0 }}
@@ -145,7 +174,11 @@ export const PromptArea: React.FC<PromptAreaProps> = ({
             onKeyDown={e => {
               if (e.key === 'Enter' && !isStreaming) handleSend()
             }}
-            placeholder="Ask me anything, add workspace files..."
+            placeholder={
+              aiMode === 'ask'
+                ? "Ask me anything, add workspace files..."
+                : "Describe the workspace you want to generate..."
+            }
           />
 
           <div className="d-flex justify-content-between">

libs/remix-ui/remix-ai-assistant/src/components/remix-ui-remix-ai-assistant.tsx

@@ -51,6 +51,7 @@ export const RemixUiRemixAiAssistant = React.forwardRef<
   const [availableModels, setAvailableModels] = useState<string[]>([])
   const [selectedModel, setSelectedModel] = useState<string | null>(null)
   const [isOllamaFailureFallback, setIsOllamaFailureFallback] = useState(false)
+  const [aiMode, setAiMode] = useState<'ask' | 'edit'>('ask')
 
   const historyRef = useRef<HTMLDivElement | null>(null)
   const modelBtnRef = useRef(null)
@@ -416,11 +417,23 @@ export const RemixUiRemixAiAssistant = React.forwardRef<
     },
     [isStreaming, props.plugin]
   )
+  const handleGenerateWorkspaceWithPrompt = useCallback(async (prompt: string) => {
+    dispatchActivity('button', 'generateWorkspace')
+    if (prompt && prompt.trim()) {
+      await sendPrompt(`/workspace ${prompt.trim()}`)
+      _paq.push(['trackEvent', 'remixAI', 'GenerateNewAIWorkspaceFromEditMode', prompt])
+    }
+  }, [sendPrompt])
 
   const handleSend = useCallback(async () => {
-    await sendPrompt(input)
+    if (aiMode === 'ask') {
+      await sendPrompt(input)
+    } else if (aiMode === 'edit') {
+      // Call generateWorkspace for edit mode with the current input
+      await handleGenerateWorkspaceWithPrompt(input)
+    }
     setInput('')
-  }, [input, sendPrompt])
+  }, [input, sendPrompt, aiMode, handleGenerateWorkspaceWithPrompt])
 
   // Added handlers for special command buttons (assumed to exist)
   const handleAddContext = useCallback(() => {
@@ -734,6 +747,8 @@ export const RemixUiRemixAiAssistant = React.forwardRef<
           aiContextGroupList={aiContextGroupList}
           aiAssistantGroupList={aiAssistantGroupList}
           textareaRef={textareaRef}
+          aiMode={aiMode}
+          setAiMode={setAiMode}
         />
       </section>
     </div>