Create Insight Generation Feature

Added integration with Messages API and Ollama

src/main.rs

@@ -31,6 +31,7 @@ use chrono::Utc;
 use diesel::sqlite::Sqlite;
 use rayon::prelude::*;
 
+use crate::ai::InsightGenerator;
 use crate::auth::login;
 use crate::data::*;
 use crate::database::models::InsertImageExif;
@@ -50,6 +51,7 @@ use log::{debug, error, info, trace, warn};
 use opentelemetry::trace::{Span, Status, TraceContextExt, Tracer};
 use opentelemetry::{KeyValue, global};
 
+mod ai;
 mod auth;
 mod data;
 mod database;
@@ -715,7 +717,7 @@ fn main() -> std::io::Result<()> {
         }
 
         create_thumbnails();
-        generate_video_gifs().await;
+        // generate_video_gifs().await;
 
         let app_data = Data::new(AppState::default());
 
@@ -744,6 +746,7 @@ fn main() -> std::io::Result<()> {
             let favorites_dao = SqliteFavoriteDao::new();
             let tag_dao = SqliteTagDao::default();
             let exif_dao = SqliteExifDao::new();
+            let insight_dao = SqliteInsightDao::new();
             let cors = Cors::default()
                 .allowed_origin_fn(|origin, _req_head| {
                     // Allow all origins in development, or check against CORS_ALLOWED_ORIGINS env var
@@ -795,6 +798,10 @@ fn main() -> std::io::Result<()> {
                 .service(delete_favorite)
                 .service(get_file_metadata)
                 .service(memories::list_memories)
+                .service(ai::generate_insight_handler)
+                .service(ai::get_insight_handler)
+                .service(ai::delete_insight_handler)
+                .service(ai::get_all_insights_handler)
                 .add_feature(add_tag_services::<_, SqliteTagDao>)
                 .app_data(app_data.clone())
                 .app_data::<Data<RealFileSystem>>(Data::new(RealFileSystem::new(
@@ -808,6 +815,10 @@ fn main() -> std::io::Result<()> {
                 .app_data::<Data<Mutex<Box<dyn ExifDao>>>>(Data::new(Mutex::new(Box::new(
                     exif_dao,
                 ))))
+                .app_data::<Data<Mutex<Box<dyn InsightDao>>>>(Data::new(Mutex::new(Box::new(
+                    insight_dao,
+                ))))
+                .app_data::<Data<InsightGenerator>>(Data::new(app_data.insight_generator.clone()))
                 .wrap(prometheus.clone())
         })
         .bind(dotenv::var("BIND_URL").unwrap())?