Screen Capture & Analysis

const handleAnalyze = async () => {
  const screenshot = await window.electron.captureScreen();
  sendMessage(
    "Analyze this coding problem. Provide Idea, Code, Walkthrough, and Test Cases.",
    { screenshot }
  );
};

const handleUpdate = async () => {
  const screenshot = await window.electron.captureScreen();
  sendMessage(
    "The interviewer added new constraints. Update the analysis for all sections.",
    { screenshot }
  );
};

const handleCodeSuggestion = async () => {
  const screenshot = await window.electron.captureScreen();
  sendMessage(
    "Suggest improvements to the current code approach. Focus on optimization and clean code.",
    { screenshot }
  );
};

// With screenshot
handleCustomPrompt(
  "Focus on the time complexity optimization for this problem",
  includeScreenshot: true
);

// Without screenshot
handleCustomPrompt(
  "Explain the trade-offs between HashMap and TreeMap here",
  includeScreenshot: false
);

export async function POST(req: Request) {
  const { messages, conversationId, screenshot } = await req.json();
  
  // Attach screenshot to the last message
  if (screenshot) {
    const lastMsg = modelMessages[modelMessages.length - 1];
    if (lastMsg && lastMsg.role === 'user') {
      lastMsg.content = [
        { type: 'text', text: lastMsg.content },
        { type: 'image', image: screenshot }
      ];
    }
  }
  
  // Stream response with structured analysis
  return streamText({
    model: myProvider.languageModel(defaultModel),
    messages: modelMessages,
    output: Output.object({ schema: analysisSchema }),
    // ...
  });
}

const analysisSchema = z.object({
  idea: z.string().describe('Problem understanding, key observations, approaches'),
  code: z.string().describe('Clean, well-commented implementation code'),
  walkthrough: z.string().describe('Step-by-step explanation of the solution'),
  testCases: z.array(z.object({
    input: z.string(),
    output: z.string(),
    reason: z.string(),
  })).describe('Edge cases and test inputs'),
  mistakes: z.array(z.object({
    mistake: z.string(),
    correction: z.string(),
    pattern: z.string(),
  })).describe('Common mistakes for this problem type'),
  memories: z.array(z.object({
    memory: z.string(),
    createdAt: z.string(),
  })).describe('Relevant memories about user preferences'),
});