ClipLLM.java

// ClipLLM.java — Build this yourself, step by step!
import java.awt.Toolkit;
import java.awt.datatransfer.Clipboard;
import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.StringSelection;
import java.lang.Thread;
import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;

public class ClipLLM {
    // Config
    static final String OLLAMA_URL = "http://localhost:11434/api/generate";
    static final String PRIMARY_MODEL = "gemma4:e4b";
    static final String FALLBACK_MODEL = "nemotron-3-nano:4b";
    static final String TRIGGER = "//ollama ";
    static final int POLL_INTERVAL_MS = 500;

    // ──── Step 1: Read Clipboard ────────────────────────────
    static String getClipboard() {
        try {
            Clipboard clipboard = Toolkit.getDefaultToolkit().getSystemClipboard();
            Object data = clipboard.getData(DataFlavor.stringFlavor);
            return data.toString();
        } catch (Exception e) {
            return "";
        }
    }

    static void setClipboard(String text) {
        try {
            StringSelection selection = new StringSelection(text);
            Clipboard clipboard = Toolkit.getDefaultToolkit().getSystemClipboard();
            clipboard.setContents(selection, null);

        } catch (Exception e) {
            System.out.println("  ⚠️  Error writing to clipboard: " + e.getMessage());
        }
    }

    // Helper to make the prompt safe for JSON (handles quotes and newlines)
    static String escapeJson(String text) {
        return text.replace("\\", "\\\\")
                .replace("\"", "\\\"")
                .replace("\n", "\\n")
                .replace("\r", "\\r")
                .replace("\t", "\\t");
    }

    static String queryOllama(String prompt, String model) {
        try {
            HttpClient client = HttpClient.newHttpClient();

            String safePrompt = escapeJson(prompt);
            String jsonBody = "{\"model\":\"" + model + "\", \"prompt\":\"" + safePrompt + "\", \"stream\":false}";

            HttpRequest request = HttpRequest.newBuilder()
                    .uri(URI.create(OLLAMA_URL))
                    .header("Content-Type", "application/json")
                    .POST(HttpRequest.BodyPublishers.ofString(jsonBody))
                    .build();

            HttpResponse<String> response = client.send(request, HttpResponse.BodyHandlers.ofString());

            if (response.statusCode() == 200) {
                String body = response.body();
                int start = body.indexOf("\"response\":\"") + 12;
                int end = body.lastIndexOf("\",\"done\":");
                if (start > 11 && end > start) {
                    String result = body.substring(start, end);
                    return result.replace("\\n", "\n")
                            .replace("\\\"", "\"")
                            .replace("\\\\", "\\")
                            .trim();
                }
                return "Error: Could not parse response";
            } else {
                return "Error: Ollama returned " + response.statusCode();
            }
        } catch (java.net.http.HttpConnectTimeoutException e) {
            return "Error: Connection timed out";
        } catch (Exception e) {
            return "Error: " + e.getMessage();
        }
    }

    static String queryWithFallback(String prompt) {
        System.out.println("  ⏳ Querying " + PRIMARY_MODEL + "...");
        String response = queryOllama(prompt, PRIMARY_MODEL);

        if (response.startsWith("Error:") || response.isEmpty()) {
            System.out.println("  ⚠️  " + PRIMARY_MODEL + " failed: " + response);
            System.out.println("  🔄 Falling back to " + FALLBACK_MODEL + "...");
            return queryOllama(prompt, FALLBACK_MODEL);
        }
        return response;
    }

    public static void main(String[] args) {
        System.out.println("\n  ╔══════════════════════════════════════════════════╗");
        printConfigLine("ClipLLM Java IS RUNNING", "");
        System.out.println("  ╠══════════════════════════════════════════════════╣");
        printConfigLine("Primary:", PRIMARY_MODEL);
        printConfigLine("Fallback:", FALLBACK_MODEL);
        printConfigLine("Trigger:", TRIGGER);
        System.out.println("  ╚══════════════════════════════════════════════════╝\n");

        String lastClipboard = getClipboard();

        while (true) {
            try {
                Thread.sleep(POLL_INTERVAL_MS);
                String currentClipboard = getClipboard();

                if (currentClipboard.equals(lastClipboard)) {
                    continue;
                }

                lastClipboard = currentClipboard;

                if (currentClipboard.startsWith(TRIGGER)) {
                    String prompt = currentClipboard.substring(TRIGGER.length()).trim();

                    if (prompt.isEmpty()) {
                        System.out.println("  Empty prompt detected.");
                        continue;
                    }

                    System.out.println(
                            "prompt: \"" + (prompt.length() > 60 ? prompt.substring(0, 57) + "..." : prompt) + "\"");

                    String result = queryWithFallback(prompt);

                    if (result.startsWith("Error:")) {
                        System.out.println("SOS: " + result);
                    } else {
                        setClipboard(result);
                        lastClipboard = result;
                        System.out.println("Done! Response copied to clipboard.");
                    }
                    System.out.println();
                }

            } catch (Exception e) {
                System.out.println("\n Stopping ClipLLM...");
                break;
            }
        }
    }

    private static void printConfigLine(String label, String value) {
        String line = String.format("  ║  %-10s %-35s ║", label, value);
        System.out.println(line);
    }
}