From 1bd0ae9b1b88e8c9fb680c7fa73565bcb944d758 Mon Sep 17 00:00:00 2001
From: Andrew Ridgway <ar17787@gmail.com>
Date: Thu, 14 Nov 2024 17:48:11 +1000
Subject: [PATCH] untested ollama

---
 src/ai_generators/__init__.py            |  0
 src/ai_generators/ollama_md_generator.py | 39 ++++++++++++++++++++++++
 src/main.py                              | 22 ++++++-------
 3 files changed, 50 insertions(+), 11 deletions(-)
 create mode 100644 src/ai_generators/__init__.py
 create mode 100644 src/ai_generators/ollama_md_generator.py

diff --git a/src/ai_generators/__init__.py b/src/ai_generators/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/ai_generators/ollama_md_generator.py b/src/ai_generators/ollama_md_generator.py
new file mode 100644
index 0000000..1717ca6
--- /dev/null
+++ b/src/ai_generators/ollama_md_generator.py
@@ -0,0 +1,39 @@
+import os
+from ollama import Client
+
+
+class OllamaGenerator:
+
+    def __init__(self, title: str, content: str):
+        self.title = title
+        self.content = content
+        self.ollama_client = Client(host=f"""
+                                    {os.getenv('OLLAMA_PROTOCOL')}://{os.getenv('OLLAMA_HOST')}:{os.getenv('OLLAMA_PORT')}
+                                    """
+                                    )
+
+    def generate_markdown(self) -> str:
+
+        prompt = f"""
+            Generate a 1000 word blog as a markdown document.
+            The title for the blog is {self.title}.
+            Do not output the title in the markdown.
+            The content for the blog is:
+                {self.content}
+            Only output markdown DO NOT GENERATE AN EXPLANATION
+            """
+        try:
+            self.response = self.ollama_client.chat(model=self.ollama_model,
+                                                    messages=[
+                    {
+                        'role': 'user',
+                        'content': f'{prompt}',
+                    },
+                ])
+
+        except Exception as e:
+            raise Exception(f"Failed to generate markdown: {e}")
+
+    def save_to_file(self, filename: str) -> None:
+        with open(filename, "w") as f:
+            f.write(self.generate_markdown())
diff --git a/src/main.py b/src/main.py
index ab9a8b7..bb345a2 100644
--- a/src/main.py
+++ b/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'])