Merge pull request 'enabling login btw users' (#5) from quart-login into main

Reviewed-on: #5

llm.py

@@ -4,9 +4,14 @@ from ollama import Client
 from openai import OpenAI
 
 import logging
+from dotenv import load_dotenv
+
+load_dotenv()
 
 logging.basicConfig(level=logging.INFO)
 
+TRY_OLLAMA = os.getenv("TRY_OLLAMA", False)
+
 
 class LLMClient:
     def __init__(self):
@@ -30,7 +35,9 @@ class LLMClient:
         prompt: str,
         system_prompt: str,
     ):
+        # Instituting a fallback if my gaming PC is not on
         if self.PROVIDER == "ollama":
+            try:
                 response = self.ollama_client.chat(
                     model="gemma3:4b",
                     messages=[
@@ -41,9 +48,11 @@ class LLMClient:
                         {"role": "user", "content": prompt},
                     ],
                 )
-            print(response)
                 output = response.message.content
-        elif self.PROVIDER == "openai":
+                return output
+            except Exception as e:
+                logging.error(f"Could not connect to OLLAMA: {str(e)}")
+
         response = self.openai_client.responses.create(
             model="gpt-4o-mini",
             input=[

main.py

@@ -197,10 +197,6 @@ def filter_indexed_files(docs):
 if __name__ == "__main__":
     args = parser.parse_args()
     if args.reindex:
-        with sqlite3.connect("./visited.db") as conn:
-            c = conn.cursor()
-            c.execute("DELETE FROM indexed_documents")
-
         logging.info("Fetching documents from Paperless-NGX")
         ppngx = PaperlessNGXService()
         docs = ppngx.get_data()