untested ollama

src/main.py

@@ -1,5 +1,4 @@
-from ollama import Client
-
+import ai_generators.ollama_md_generator as omg
 import trilium.notes as tn
 
 tril = tn.TrilumNotes()
@@ -7,17 +6,18 @@ tril = tn.TrilumNotes()
 tril.get_new_notes()
 tril_notes = tril.get_notes_content()
 
+def convert_to_lowercase_with_underscores(string):
+    return string.lower().replace(" ", "_")
+
+
 for note in tril_notes:
     print(tril_notes[note]['title'])
     print(tril_notes[note]['content'])
-client = Client(host='http://192.168.178.45:11434')
+    print("Generating Document")
+    ai_gen = omg.OllamaGenerator(tril_notes[note]['title'], 
+                                 tril_notes[note]['content'])
+    os_friendly_title = convert_to_lowercase_with_underscores(tril_notes[note]['title'])
+    omg.save_to_file(f"{os_friendly_title}.md")
+
 
-user_prompt = input("Ask mistral-nemo a question: ")
 
-response = client.chat(model='mistral-nemo', messages=[
-  {
-    'role': 'user',
-    'content': f'{user_prompt}',
-  },
-])
-print(response['message']['content'])