Compare commits
5 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 8870741c3e | |||
| 57133f3d2c | |||
| d7fff8a58a | |||
| d0998ec50a | |||
| e26b1de89c |
@@ -0,0 +1,2 @@
|
||||
main
|
||||
.cache
|
||||
@@ -0,0 +1,14 @@
|
||||
CC = gcc
|
||||
CFLAGS = -Wall -g -Iinclude
|
||||
TARGET = main
|
||||
SRC = main.c deepgram_client.c utils.c intelligence.c audio.c
|
||||
LIBS = -lavdevice -lavformat -lavcodec -lavutil -lasound -lcurl
|
||||
|
||||
$(TARGET): $(SRC)
|
||||
$(CC) $(CFLAGS) -o $(TARGET) $(SRC) $(LIBS)
|
||||
|
||||
clean:
|
||||
rm -f *.o $(TARGET)
|
||||
|
||||
install-deps:
|
||||
sudo apt install build-essential libavformat-dev libavdevice-dev libavcodec-dev libavutil-dev libasound2-dev
|
||||
@@ -0,0 +1,34 @@
|
||||
# Conversation
|
||||
This proof of concept is for implementing conversational loop with AI at a low level with raw C.
|
||||
|
||||
## How it works
|
||||
This simple program:
|
||||
1. Captures audio samples for x seconds into a local file using alsa
|
||||
2. Sends the raw pcm data to Deepgram STT API for transcription
|
||||
3. Uses the transcript to call Anthropic to get an assistant's response
|
||||
4. Converts Anthropic's response to audio using Deepgram's TTS
|
||||
5. Plays back the generated audio from Deepgram using alsa
|
||||
6. Quits. _There is no conversational loop at the time of this writing._
|
||||
|
||||
## Requirements
|
||||
Platform: the project is meant for Linux x64. It was successfully run on arch linux, with no guarantees otherwise.
|
||||
|
||||
* libcurl
|
||||
* build-essential
|
||||
* libasound2-dev
|
||||
|
||||
_See more deps in Makefile `install-deps` command._
|
||||
|
||||
## Getting started
|
||||
1. Add `DEEPGRAM_API_KEY` and `ANTHROPIC_API_KEY` to `.bashrc` as environment variables.
|
||||
2. Build & run
|
||||
```sh
|
||||
make
|
||||
./main
|
||||
```
|
||||
|
||||
## Audio configuration
|
||||
Please see the `audio.h` and `audio.c` as the params for alsa were hardcoded based on my local arch linux x64 machine. You may have to adjust it based on your setup (e.g. you may have a mic with 2 channels/stereo, which would change the size of each frame).
|
||||
|
||||
## Future work
|
||||
Add an event loop, maybe add multi-threading, so that we can interrupt the AI, ask follow-up questions, and make it sound like a back and forth conversation.
|
||||
@@ -0,0 +1,215 @@
|
||||
#include "audio.h"
|
||||
|
||||
static void print_pcm_state(snd_pcm_t *pcm_handle) {
|
||||
snd_pcm_state_t state = snd_pcm_state(pcm_handle);
|
||||
const char *state_name = snd_pcm_state_name(state);
|
||||
printf("State of pcm handle: %s\n", state_name);
|
||||
}
|
||||
|
||||
int audio_play(char *buf, size_t byte_count) {
|
||||
printf("going to play audio of %d bytes of data\n", (int)byte_count);
|
||||
|
||||
snd_pcm_t *pcm_handle;
|
||||
// The total frames in the buffer is how many bytes divided by 2 because
|
||||
// of linear16 encoding, one channel
|
||||
size_t total_frames = byte_count / 2;
|
||||
|
||||
int err;
|
||||
err = snd_pcm_open(&pcm_handle, ALSA_SPEAKER_HARDWARE_DEVICE,
|
||||
SND_PCM_STREAM_PLAYBACK, 0);
|
||||
if (err < 0) {
|
||||
fprintf(stderr, "unable to open playback device");
|
||||
return err;
|
||||
}
|
||||
print_pcm_state(pcm_handle);
|
||||
|
||||
const char *name = snd_pcm_name(pcm_handle);
|
||||
printf("found this device: %s\n", name);
|
||||
|
||||
snd_pcm_info_t *pcm_info;
|
||||
snd_pcm_info_malloc(&pcm_info);
|
||||
err = snd_pcm_info(pcm_handle, pcm_info);
|
||||
if (err < 0) {
|
||||
snd_pcm_close(pcm_handle);
|
||||
printf("unable to get info of pcm device");
|
||||
return -1;
|
||||
}
|
||||
name = snd_pcm_info_get_name(pcm_info);
|
||||
if (err < 0) {
|
||||
printf("unable to get card info");
|
||||
snd_pcm_close(pcm_handle);
|
||||
snd_pcm_info_free(pcm_info);
|
||||
return -1;
|
||||
}
|
||||
printf("got pcm name: %s \n", name);
|
||||
|
||||
print_pcm_state(pcm_handle);
|
||||
snd_pcm_info_free(pcm_info);
|
||||
if ((err = snd_pcm_set_params(pcm_handle, SND_PCM_FORMAT_S16_LE,
|
||||
SND_PCM_ACCESS_RW_INTERLEAVED, 1, 48000, 1,
|
||||
500000)) < 0) { /* 0.5sec */
|
||||
printf("Playback open error: %s\n", snd_strerror(err));
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
// we want to read in a chunk of data and writei to the pcm
|
||||
size_t frames_written = 0;
|
||||
size_t frames_in_each_chunk = 1024; // how many frames to read per chunk (2048
|
||||
// bytes since each frame is 2 bytes)
|
||||
while (frames_written < total_frames) {
|
||||
size_t frames_to_write =
|
||||
frames_in_each_chunk < (total_frames - frames_written)
|
||||
? frames_in_each_chunk
|
||||
: (total_frames - frames_written);
|
||||
// We move the pointer x2 because each frame is 2 bytes so the next audio
|
||||
// sample is 16 bits forward, not 8
|
||||
char *start_ptr = buf + (frames_written * 2);
|
||||
snd_pcm_sframes_t result =
|
||||
snd_pcm_writei(pcm_handle, start_ptr, frames_to_write);
|
||||
if (result < 0) {
|
||||
result = snd_pcm_recover(pcm_handle, result, 0);
|
||||
if (result < 0) {
|
||||
printf("snd_pcm_writei failed: %s\n", snd_strerror(result));
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (result > 0 && result < frames_to_write) {
|
||||
printf("Short write \n");
|
||||
}
|
||||
|
||||
frames_written += result;
|
||||
}
|
||||
|
||||
/* pass the remaining samples, otherwise they're dropped in close */
|
||||
err = snd_pcm_drain(pcm_handle);
|
||||
if (err < 0) {
|
||||
printf("snd_pcm_drain failed: %s\n", snd_strerror(err));
|
||||
}
|
||||
snd_pcm_close(pcm_handle);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
char *audio_capture(uint seconds) {
|
||||
char *output_file = "output.pcm";
|
||||
|
||||
snd_pcm_t *pcm_handle = NULL;
|
||||
|
||||
int err;
|
||||
err = snd_pcm_open(&pcm_handle, ALSA_MIC_HARDWARE_DEVICE,
|
||||
SND_PCM_STREAM_CAPTURE, 0);
|
||||
if (err < 0) {
|
||||
printf("unable to open pcm device");
|
||||
return NULL;
|
||||
}
|
||||
print_pcm_state(pcm_handle);
|
||||
|
||||
const char *name = snd_pcm_name(pcm_handle);
|
||||
printf("found this device: %s\n", name);
|
||||
|
||||
snd_pcm_info_t *pcm_info;
|
||||
snd_pcm_info_malloc(&pcm_info);
|
||||
err = snd_pcm_info(pcm_handle, pcm_info);
|
||||
if (err < 0) {
|
||||
snd_pcm_close(pcm_handle);
|
||||
printf("unable to get info of pcm device");
|
||||
return NULL;
|
||||
}
|
||||
name = snd_pcm_info_get_name(pcm_info);
|
||||
if (err < 0) {
|
||||
printf("unable to get card info");
|
||||
snd_pcm_close(pcm_handle);
|
||||
snd_pcm_info_free(pcm_info);
|
||||
return NULL;
|
||||
}
|
||||
printf("got pcm name: %s \n", name);
|
||||
|
||||
print_pcm_state(pcm_handle);
|
||||
snd_pcm_info_free(pcm_info);
|
||||
|
||||
snd_pcm_hw_params_t *params;
|
||||
err = snd_pcm_hw_params_malloc(¶ms);
|
||||
if (err < 0) {
|
||||
fprintf(stderr, "unable to mallow hw params");
|
||||
return NULL;
|
||||
}
|
||||
err = snd_pcm_hw_params_any(pcm_handle, params);
|
||||
if (err < 0) {
|
||||
fprintf(stderr, "unable to find ranges for hw device");
|
||||
snd_pcm_close(pcm_handle);
|
||||
snd_pcm_hw_params_free(params);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
uint period_size = 1024;
|
||||
uint buffer_size = 4096;
|
||||
uint channel_count = 1;
|
||||
uint bytes_per_sample = 2; // FORMAT_S16_LE
|
||||
uint samples_per_second = 48000;
|
||||
snd_pcm_hw_params_set_access(pcm_handle, params,
|
||||
SND_PCM_ACCESS_RW_INTERLEAVED);
|
||||
snd_pcm_hw_params_set_channels(pcm_handle, params, channel_count);
|
||||
snd_pcm_hw_params_set_buffer_size(pcm_handle, params, buffer_size);
|
||||
snd_pcm_hw_params_set_format(pcm_handle, params, SND_PCM_FORMAT_S16_LE);
|
||||
snd_pcm_hw_params_set_rate(pcm_handle, params, samples_per_second, 0);
|
||||
snd_pcm_hw_params_set_periods(pcm_handle, params, period_size, 0);
|
||||
|
||||
err = snd_pcm_hw_params(pcm_handle, params);
|
||||
if (err < 0) {
|
||||
fprintf(stderr, "unable to prepare pcm\n");
|
||||
snd_pcm_close(pcm_handle);
|
||||
snd_pcm_hw_params_free(params);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
print_pcm_state(pcm_handle);
|
||||
|
||||
err = snd_pcm_start(pcm_handle);
|
||||
if (err < 0) {
|
||||
fprintf(stderr, "unable to start pcm");
|
||||
snd_pcm_close(pcm_handle);
|
||||
snd_pcm_hw_params_free(params);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
print_pcm_state(pcm_handle);
|
||||
|
||||
const uint seconds_to_capture = seconds;
|
||||
const uint periods_to_read =
|
||||
(samples_per_second * seconds_to_capture) / period_size;
|
||||
char *buf = malloc(period_size * channel_count * bytes_per_sample);
|
||||
uint periods_read = 0;
|
||||
/* Read one period at a time from the hardware buffer
|
||||
* We set the hardware buffer to read 4096 frames.
|
||||
* The period size is 1024 frames.
|
||||
* Each period is 2048 BYTES because each frame is 2 bytes.
|
||||
* Each frame is 2 bytes because of 1 channel, and 16 bit format
|
||||
*/
|
||||
FILE *file = fopen(output_file, "wb");
|
||||
if (file == NULL) {
|
||||
perror("Error opening file");
|
||||
return NULL;
|
||||
}
|
||||
printf("======READING FRAMES for 5 seconds========\n");
|
||||
while (snd_pcm_readi(pcm_handle, (void *)buf, period_size) > 0 &&
|
||||
periods_read < periods_to_read) {
|
||||
// Because each frame is 2 bytes (format/bit rate), cast to int16
|
||||
// int16_t *samples = (int16_t *)buf;
|
||||
// for (int i = 0; i < period_size; i++) {
|
||||
// printf("%d ", samples[i]);
|
||||
// }
|
||||
|
||||
fwrite(buf, sizeof(int16_t), period_size, file);
|
||||
periods_read++;
|
||||
}
|
||||
printf("done capturing audio\n");
|
||||
|
||||
fclose(file);
|
||||
|
||||
free(buf);
|
||||
snd_pcm_close(pcm_handle);
|
||||
snd_pcm_hw_params_free(params);
|
||||
|
||||
return output_file;
|
||||
}
|
||||
@@ -0,0 +1,16 @@
|
||||
[
|
||||
{
|
||||
"arguments": [
|
||||
"/usr/bin/gcc",
|
||||
"-c",
|
||||
"-Wall",
|
||||
"-g",
|
||||
"-o",
|
||||
"audio_capture",
|
||||
"audio_capture.c"
|
||||
],
|
||||
"directory": "/home/talksik/Documents/research_and_development/ai-conversation",
|
||||
"file": "/home/talksik/Documents/research_and_development/ai-conversation/audio_capture.c",
|
||||
"output": "/home/talksik/Documents/research_and_development/ai-conversation/audio_capture"
|
||||
}
|
||||
]
|
||||
@@ -0,0 +1,168 @@
|
||||
#include "deepgram_client.h"
|
||||
#include <string.h>
|
||||
|
||||
// Returns string that caller must free
|
||||
static char *dg_find_transcript(char *json_response) {
|
||||
printf("Parsing response: %s \n", json_response);
|
||||
char *start = strstr(json_response, "\"transcript\":\"");
|
||||
if (!start)
|
||||
return NULL;
|
||||
|
||||
start += 14; // Skip past "transcript":"
|
||||
char *end = strchr(start, '"'); // Find closing quote
|
||||
// Copy substring between start and end
|
||||
size_t len = end - start;
|
||||
char *result = malloc(len + 1);
|
||||
strncpy(result, start, len);
|
||||
result[len] = '\0';
|
||||
return result;
|
||||
}
|
||||
|
||||
char *dg_transcribe(FILE *audio_file) {
|
||||
CURL *curl;
|
||||
CURLcode res;
|
||||
struct curl_slist *headers = NULL;
|
||||
|
||||
// Initialize response buffer
|
||||
struct curl_response_data response = {0};
|
||||
response.data = malloc(1);
|
||||
response.size = 0;
|
||||
|
||||
// Get file size
|
||||
fseek(audio_file, 0, SEEK_END);
|
||||
long file_size = ftell(audio_file);
|
||||
fseek(audio_file, 0, SEEK_SET);
|
||||
|
||||
struct curl_file_data data;
|
||||
data.file = audio_file;
|
||||
data.size = file_size;
|
||||
|
||||
curl = curl_easy_init();
|
||||
if (curl) {
|
||||
// Set headers for Deepgram API
|
||||
char *deepgram_api_key = getenv("DEEPGRAM_API_KEY");
|
||||
if (!deepgram_api_key) {
|
||||
fprintf(stderr, "DEEPGRAM_API_KEY environment variable not set\n");
|
||||
free(response.data);
|
||||
curl_easy_cleanup(curl);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
char auth_header[256];
|
||||
snprintf(auth_header, sizeof(auth_header), "Authorization: Token %s", deepgram_api_key);
|
||||
headers = curl_slist_append(headers, auth_header);
|
||||
|
||||
curl_easy_setopt(curl, CURLOPT_URL,
|
||||
"https://api.deepgram.com/v1/"
|
||||
"listen?encoding=linear16&sample_rate=48000&channels=1");
|
||||
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
|
||||
curl_easy_setopt(curl, CURLOPT_POST, 1L);
|
||||
curl_easy_setopt(curl, CURLOPT_READFUNCTION, curl_read_callback_file);
|
||||
curl_easy_setopt(curl, CURLOPT_READDATA, &data);
|
||||
curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, file_size);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, curl_write_callback_response);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &response);
|
||||
|
||||
res = curl_easy_perform(curl);
|
||||
if (res != CURLE_OK) {
|
||||
fprintf(stderr, "curl failed: %s\n", curl_easy_strerror(res));
|
||||
free(response.data);
|
||||
curl_slist_free_all(headers);
|
||||
curl_easy_cleanup(curl);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
printf("done performing curl\n");
|
||||
|
||||
curl_slist_free_all(headers);
|
||||
curl_easy_cleanup(curl);
|
||||
}
|
||||
|
||||
// Parse transcript from complete response
|
||||
char *transcript = dg_find_transcript(response.data);
|
||||
free(response.data);
|
||||
|
||||
return transcript;
|
||||
}
|
||||
|
||||
struct curl_response_data *dg_text_to_speech(char *text) {
|
||||
CURL *curl;
|
||||
CURLcode res;
|
||||
struct curl_slist *headers = NULL;
|
||||
|
||||
// Create JSON payload - need to escape the text properly
|
||||
// Worst case: every char needs escaping, so allocate 2x + overhead
|
||||
size_t text_len = strlen(text);
|
||||
size_t payload_size = text_len * 2 + 100;
|
||||
char *json_payload = malloc(payload_size);
|
||||
|
||||
// Simple approach: replace problematic chars with spaces
|
||||
char *escaped_text = malloc(text_len + 1);
|
||||
strcpy(escaped_text, text);
|
||||
for (int i = 0; escaped_text[i]; i++) {
|
||||
if (escaped_text[i] == '"' || escaped_text[i] == '\n' ||
|
||||
escaped_text[i] == '\r' || escaped_text[i] == '\t') {
|
||||
escaped_text[i] = ' ';
|
||||
}
|
||||
}
|
||||
|
||||
snprintf(json_payload, payload_size,
|
||||
"{"
|
||||
"\"text\":\"%s\""
|
||||
"}",
|
||||
escaped_text);
|
||||
|
||||
free(escaped_text);
|
||||
|
||||
// Initialize response buffer
|
||||
struct curl_response_data *response =
|
||||
malloc(sizeof(struct curl_response_data));
|
||||
response->data = malloc(1);
|
||||
response->size = 0;
|
||||
|
||||
curl = curl_easy_init();
|
||||
if (curl) {
|
||||
char *deepgram_api_key = getenv("DEEPGRAM_API_KEY");
|
||||
if (!deepgram_api_key) {
|
||||
fprintf(stderr, "DEEPGRAM_API_KEY environment variable not set\n");
|
||||
free(response->data);
|
||||
curl_easy_cleanup(curl);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
char auth_header[256];
|
||||
snprintf(auth_header, sizeof(auth_header), "Authorization: Token %s", deepgram_api_key);
|
||||
headers = curl_slist_append(headers, auth_header);
|
||||
headers = curl_slist_append(headers, "Content-Type: application/json");
|
||||
|
||||
curl_easy_setopt(curl, CURLOPT_URL,
|
||||
"https://api.deepgram.com/v1/"
|
||||
"speak?encoding=linear16&sample_rate=48000");
|
||||
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
|
||||
curl_easy_setopt(curl, CURLOPT_POST, 1L);
|
||||
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, json_payload);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, curl_write_callback_response);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEDATA, response);
|
||||
|
||||
res = curl_easy_perform(curl);
|
||||
if (res != CURLE_OK) {
|
||||
fprintf(stderr, "curl failed: %s\n", curl_easy_strerror(res));
|
||||
free(response->data);
|
||||
free(response);
|
||||
curl_slist_free_all(headers);
|
||||
curl_easy_cleanup(curl);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
curl_slist_free_all(headers);
|
||||
curl_easy_cleanup(curl);
|
||||
}
|
||||
|
||||
printf("received audio of %d length for tts\n", (int)response->size);
|
||||
|
||||
// Debug: print first 100 chars to see if it's audio or error message
|
||||
printf("First 100 chars: %.100s\n", response->data);
|
||||
|
||||
free(json_payload);
|
||||
return response;
|
||||
}
|
||||
@@ -0,0 +1,29 @@
|
||||
#ifndef AUDIOCAPTURE_H
|
||||
#define AUDIOCAPTURE_H
|
||||
|
||||
#include <stdint.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <alsa/asoundlib.h>
|
||||
#include <alsa/control.h>
|
||||
#include <alsa/pcm.h>
|
||||
|
||||
// NOTE: find this via `arecord -l`
|
||||
#define ALSA_MIC_HARDWARE_DEVICE "hw:3"
|
||||
#define ALSA_SPEAKER_HARDWARE_DEVICE "default"
|
||||
|
||||
// Captures audio for given seconds, and returns file name that we captured the
|
||||
// audio into
|
||||
char *audio_capture(uint seconds);
|
||||
|
||||
/* count: how many bytes in the buffer
|
||||
* returns int: 0 if success, -1 if failure
|
||||
* NOTE: This function assumes that the audio data is using:
|
||||
* - linear16 formatting (each sample is 16 bits)
|
||||
* - one channel (1 sample)
|
||||
* - with 48000 HZ sampling rate
|
||||
* Hence, it's 2 bytes per frame.
|
||||
*/
|
||||
int audio_play(char *buf, size_t byte_count);
|
||||
|
||||
#endif // AUDIOCAPTURE_H
|
||||
@@ -0,0 +1,21 @@
|
||||
#ifndef DEEPGRAMCLIENT_H
|
||||
#define DEEPGRAMCLIENT_H
|
||||
|
||||
#include <curl/curl.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include "utils.h"
|
||||
|
||||
/*
|
||||
* @param audio_file: must already be opened
|
||||
* @returns char *: transcript string result. Call must free.
|
||||
*/
|
||||
char *dg_transcribe(FILE *audio_file);
|
||||
|
||||
/*
|
||||
* @returns response data. Caller must free the struct AND the data within it.
|
||||
* TODO: should return custom struct with details about encoding, sample rate, etc.
|
||||
*/
|
||||
struct curl_response_data *dg_text_to_speech(char *text);
|
||||
|
||||
#endif // DEEPGRAMCLIENT_H
|
||||
@@ -0,0 +1,9 @@
|
||||
#ifndef INTELLIGENCE_H
|
||||
#define INTELLIGENCE_H
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
char *ai_ask(char *question);
|
||||
|
||||
#endif //INTELLIGENCE_H
|
||||
@@ -0,0 +1,24 @@
|
||||
#ifndef UTILS_H
|
||||
#define UTILS_H
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
|
||||
struct curl_file_data {
|
||||
FILE *file;
|
||||
size_t size;
|
||||
};
|
||||
|
||||
struct curl_response_data {
|
||||
char *data;
|
||||
size_t size;
|
||||
};
|
||||
|
||||
// This callback assumes the userdata is `struct file_data`
|
||||
size_t curl_read_callback_file(void *ptr, size_t size, size_t nmemb, void *userdata);
|
||||
|
||||
// This callback assumes that we are aggregating callback data into an allocated response_data struct
|
||||
size_t curl_write_callback_response(void *contents, size_t size, size_t nmemb,
|
||||
void *userp);
|
||||
|
||||
#endif // UTILS_H
|
||||
@@ -0,0 +1,86 @@
|
||||
#include "intelligence.h"
|
||||
#include "utils.h"
|
||||
#include <curl/curl.h>
|
||||
#include <string.h>
|
||||
|
||||
static char *find_ai_response(char *json_response) {
|
||||
printf("Parsing response: %s \n", json_response);
|
||||
char *start = strstr(json_response, "\"text\":\"");
|
||||
if (!start)
|
||||
return NULL;
|
||||
|
||||
start += 8; // Skip past "text":"
|
||||
char *end = strchr(start, '"');
|
||||
if (!end)
|
||||
return NULL;
|
||||
|
||||
size_t len = end - start;
|
||||
char *result = malloc(len + 1);
|
||||
strncpy(result, start, len);
|
||||
result[len] = '\0';
|
||||
return result;
|
||||
}
|
||||
|
||||
char *ai_ask(char *question) {
|
||||
CURL *curl;
|
||||
CURLcode res;
|
||||
struct curl_slist *headers = NULL;
|
||||
|
||||
// Initialize response buffer
|
||||
struct curl_response_data response = {0};
|
||||
response.data = malloc(1);
|
||||
response.size = 0;
|
||||
|
||||
// Create JSON payload
|
||||
char json_payload[2048];
|
||||
snprintf(json_payload, sizeof(json_payload),
|
||||
"{"
|
||||
"\"model\":\"claude-3-haiku-20240307\","
|
||||
"\"max_tokens\":256,"
|
||||
"\"messages\":[{\"role\":\"user\",\"content\":\"Make it one sentence answer: %s\"}]"
|
||||
"}",
|
||||
question);
|
||||
|
||||
curl = curl_easy_init();
|
||||
if (curl) {
|
||||
char *anthropic_api_key = getenv("ANTHROPIC_API_KEY");
|
||||
if (!anthropic_api_key) {
|
||||
printf("ANTHROPIC_API_KEY not set\n");
|
||||
free(response.data);
|
||||
curl_easy_cleanup(curl);
|
||||
return NULL;
|
||||
}
|
||||
char auth_header[256];
|
||||
snprintf(auth_header, sizeof(auth_header), "x-api-key: %s",
|
||||
anthropic_api_key);
|
||||
headers = curl_slist_append(headers, "content-type: application/json");
|
||||
headers = curl_slist_append(headers, auth_header);
|
||||
headers = curl_slist_append(headers, "anthropic-version: 2023-06-01");
|
||||
|
||||
curl_easy_setopt(curl, CURLOPT_URL,
|
||||
"https://api.anthropic.com/v1/messages");
|
||||
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
|
||||
curl_easy_setopt(curl, CURLOPT_POST, 1L);
|
||||
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, json_payload);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, curl_write_callback_response);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &response);
|
||||
|
||||
res = curl_easy_perform(curl);
|
||||
if (res != CURLE_OK) {
|
||||
fprintf(stderr, "curl failed: %s\n", curl_easy_strerror(res));
|
||||
free(response.data);
|
||||
curl_slist_free_all(headers);
|
||||
curl_easy_cleanup(curl);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
curl_slist_free_all(headers);
|
||||
curl_easy_cleanup(curl);
|
||||
}
|
||||
|
||||
// Parse AI response from complete response
|
||||
char *answer = find_ai_response(response.data);
|
||||
free(response.data);
|
||||
|
||||
return answer;
|
||||
}
|
||||
Executable
BIN
Binary file not shown.
@@ -0,0 +1,45 @@
|
||||
#include "audio.h"
|
||||
#include "deepgram_client.h"
|
||||
#include "intelligence.h"
|
||||
#include "utils.h"
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
int main(int argc, char *argv[]) {
|
||||
char *capture_audio_file = audio_capture(5);
|
||||
|
||||
// Reopen file for reading and transcribe
|
||||
FILE *file = fopen(capture_audio_file, "rb");
|
||||
if (file != NULL) {
|
||||
char *transcript = dg_transcribe(file);
|
||||
if (transcript) {
|
||||
printf("Transcript: %s\n", transcript);
|
||||
|
||||
char *ai_answer = ai_ask(transcript);
|
||||
if (ai_answer) {
|
||||
printf("AI answer: %s\n", ai_answer);
|
||||
|
||||
struct curl_response_data *tts_response = dg_text_to_speech(ai_answer);
|
||||
int result = audio_play(tts_response->data, tts_response->size);
|
||||
if (result != 0) {
|
||||
fprintf(stderr, "failure in playback");
|
||||
}
|
||||
|
||||
free(tts_response->data);
|
||||
free(tts_response);
|
||||
|
||||
// we have this response data, and buffer within it. we can loop through
|
||||
// it and write the pcm data within it to alsa
|
||||
free(ai_answer);
|
||||
}
|
||||
free(transcript);
|
||||
} else {
|
||||
fprintf(stderr, "unable to get transcript");
|
||||
}
|
||||
fclose(file);
|
||||
} else {
|
||||
fprintf(stderr, "unable to open file for reading\n");
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
Binary file not shown.
@@ -0,0 +1,30 @@
|
||||
#include "utils.h"
|
||||
#include <string.h>
|
||||
|
||||
size_t curl_read_callback_file(void *ptr, size_t size, size_t nmemb,
|
||||
void *userdata) {
|
||||
struct curl_file_data *data = (struct curl_file_data *)userdata;
|
||||
size_t bytes_to_read = size * nmemb;
|
||||
return fread(ptr, 1, bytes_to_read, data->file);
|
||||
}
|
||||
|
||||
size_t curl_write_callback_response(void *contents, size_t size, size_t nmemb,
|
||||
void *userp) {
|
||||
size_t total_size = size * nmemb;
|
||||
struct curl_response_data *response = (struct curl_response_data *)userp;
|
||||
|
||||
// Reallocate buffer to fit new data
|
||||
response->data = realloc(response->data, response->size + total_size + 1);
|
||||
if (response->data == NULL) {
|
||||
return 0; // Error
|
||||
}
|
||||
|
||||
// Copy new data to buffer
|
||||
memcpy(&response->data[response->size], contents, total_size);
|
||||
response->size += total_size;
|
||||
response->data[response->size] = '\0'; // Null terminate
|
||||
|
||||
printf("read an extra %d bytes of data\n", (int)total_size);
|
||||
|
||||
return total_size;
|
||||
}
|
||||
@@ -0,0 +1,2 @@
|
||||
# computeruse
|
||||
This is a prototype.
|
||||
@@ -6,6 +6,7 @@ import pyautogui
|
||||
import subprocess
|
||||
from io import BytesIO
|
||||
from PIL import Image
|
||||
import quartz_doubleclick
|
||||
|
||||
ANTHROPIC_API_URL = "https://api.anthropic.com/v1/messages"
|
||||
MODEL = "claude-3-7-sonnet-20250219"
|
||||
@@ -103,7 +104,8 @@ def execute_computer_tool(tool_input):
|
||||
return {"type": "text", "text": "Error: Missing coordinates for double click action"}
|
||||
|
||||
# Perform double click
|
||||
pyautogui.doubleClick(x, y, interval=0.2)
|
||||
# pyautogui.doubleClick(x, y, interval=0.2) this doesn't work
|
||||
quartz_doubleclick.double_click(x, y)
|
||||
return {"type": "text", "text": f"Double-clicked at coordinates ({x}, {y})"}
|
||||
|
||||
elif action == "type":
|
||||
@@ -114,19 +116,24 @@ def execute_computer_tool(tool_input):
|
||||
|
||||
elif action == "key":
|
||||
# Press a key or key combination
|
||||
key = tool_input.get("key", "")
|
||||
text = tool_input.get("text", "")
|
||||
try:
|
||||
pyautogui.press(key)
|
||||
return {"type": "text", "text": f"Pressed key: {key}"}
|
||||
if '+' in text:
|
||||
# Handle key combinations like "command+c"
|
||||
keys = text.replace('super', 'command').split('+')
|
||||
pyautogui.hotkey(*keys, interval=0.05) # interval is required
|
||||
else:
|
||||
pyautogui.press(text)
|
||||
return {"type": "text", "text": f"Pressed key: {text}"}
|
||||
except Exception as e:
|
||||
return {"type": "text", "text": f"Error pressing key {key}: {str(e)}"}
|
||||
return {"type": "text", "text": f"Error pressing key {text}: {str(e)}"}
|
||||
|
||||
elif action == "scroll":
|
||||
# Scroll action
|
||||
direction = tool_input.get("direction", "down")
|
||||
amount = tool_input.get("amount", 3)
|
||||
# Scroll action (should we really have defaults here?)
|
||||
direction = tool_input.get("scroll_direction", "down")
|
||||
amount = tool_input.get("scroll_amount", 3)
|
||||
|
||||
scroll_amount = -amount if direction == "up" else amount
|
||||
scroll_amount = -amount if direction == "down" else amount
|
||||
pyautogui.scroll(scroll_amount)
|
||||
return {"type": "text", "text": f"Scrolled {direction} by {amount}"}
|
||||
|
||||
@@ -158,23 +165,23 @@ def execute_bash_tool(command):
|
||||
return {"type": "text", "text": f"Error executing command: {str(e)}"}
|
||||
|
||||
|
||||
def execute_text_editor_tool(command, path, **kwargs):
|
||||
def execute_text_editor_tool(_command, _path, **kwargs):
|
||||
"""Execute text editor actions."""
|
||||
|
||||
if command == "view":
|
||||
if _command == "view":
|
||||
try:
|
||||
with open(path, 'r') as f:
|
||||
with open(_path, 'r') as f:
|
||||
content = f.read()
|
||||
return {"type": "text", "text": content}
|
||||
except Exception as e:
|
||||
return {"type": "text", "text": f"Error reading file: {str(e)}"}
|
||||
|
||||
elif command == "str_replace":
|
||||
elif _command == "str_replace":
|
||||
old_str = kwargs.get("old_str", "")
|
||||
new_str = kwargs.get("new_str", "")
|
||||
|
||||
try:
|
||||
with open(path, 'r') as f:
|
||||
with open(_path, 'r') as f:
|
||||
content = f.read()
|
||||
|
||||
if old_str not in content:
|
||||
@@ -184,7 +191,7 @@ def execute_text_editor_tool(command, path, **kwargs):
|
||||
new_content = content.replace(old_str, new_str)
|
||||
|
||||
# Write back to file
|
||||
with open(path, 'w') as f:
|
||||
with open(_path, 'w') as f:
|
||||
f.write(new_content)
|
||||
|
||||
return {"type": "text", "text": "String replaced successfully"}
|
||||
@@ -192,17 +199,17 @@ def execute_text_editor_tool(command, path, **kwargs):
|
||||
except Exception as e:
|
||||
return {"type": "text", "text": f"Error modifying file: {str(e)}"}
|
||||
|
||||
elif command == "create":
|
||||
elif _command == "create":
|
||||
content = kwargs.get("content", "")
|
||||
try:
|
||||
with open(path, 'w') as f:
|
||||
with open(_path, 'w') as f:
|
||||
f.write(content)
|
||||
return {"type": "text", "text": f"File created: {path}"}
|
||||
return {"type": "text", "text": f"File created: {_path}"}
|
||||
except Exception as e:
|
||||
return {"type": "text", "text": f"Error creating file: {str(e)}"}
|
||||
|
||||
else:
|
||||
return {"type": "text", "text": f"Unknown text editor command: {command}"}
|
||||
return {"type": "text", "text": f"Unknown text editor command: {_command}"}
|
||||
|
||||
|
||||
def execute_tool(tool_name, tool_input):
|
||||
@@ -0,0 +1,39 @@
|
||||
'''
|
||||
This script simulates a double-click at the given mouse cursor position.
|
||||
'''
|
||||
|
||||
import Quartz
|
||||
from time import sleep
|
||||
|
||||
def post_mouse_event(type, pos, click_state):
|
||||
event = Quartz.CGEventCreateMouseEvent(
|
||||
None, type, pos, Quartz.kCGMouseButtonLeft
|
||||
)
|
||||
Quartz.CGEventSetIntegerValueField(event, Quartz.kCGMouseEventClickState, click_state)
|
||||
Quartz.CGEventPost(Quartz.kCGHIDEventTap, event)
|
||||
|
||||
def double_click (x, y):
|
||||
pos = None
|
||||
if x is None or y is None:
|
||||
# Get current mouse position if no coordinates are provided
|
||||
loc = Quartz.CGEventGetLocation(Quartz.CGEventCreate(None))
|
||||
pos = (loc.x, loc.y)
|
||||
else:
|
||||
# Use provided coordinates
|
||||
pos = (x, y)
|
||||
|
||||
# First click
|
||||
post_mouse_event(Quartz.kCGEventLeftMouseDown, pos, 1)
|
||||
post_mouse_event(Quartz.kCGEventLeftMouseUp, pos, 1)
|
||||
|
||||
sleep(0.05) # Short delay within double-click threshold
|
||||
|
||||
# Second click
|
||||
post_mouse_event(Quartz.kCGEventLeftMouseDown, pos, 2)
|
||||
post_mouse_event(Quartz.kCGEventLeftMouseUp, pos, 2)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
print('test: double clicking at cursor position in 3 seconds...')
|
||||
sleep(3)
|
||||
double_click()
|
||||
Reference in New Issue
Block a user