From 23d77ee33879ee71caa2764650bbb8ed82f74ca9 Mon Sep 17 00:00:00 2001
From: sabaimran <65192171+sabaimran@users.noreply.github.com>
Date: Wed, 26 Jul 2023 22:45:52 +0000
Subject: [PATCH] Fix import issues in desktop image builds (#343)

---
 Khoj.spec        | 3 +++
 src/khoj/main.py | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/Khoj.spec b/Khoj.spec
index a2d92e29..54069c4a 100644
--- a/Khoj.spec
+++ b/Khoj.spec
@@ -19,6 +19,9 @@ datas += copy_metadata('filelock')
 datas += copy_metadata('numpy')
 datas += copy_metadata('tokenizers')
 datas += copy_metadata('pillow')
+datas += copy_metadata('huggingface_hub')
+datas += copy_metadata('safetensors')
+datas += copy_metadata('pyyaml')
 
 block_cipher = None
 
diff --git a/src/khoj/main.py b/src/khoj/main.py
index 6689805c..22bb8586 100644
--- a/src/khoj/main.py
+++ b/src/khoj/main.py
@@ -75,7 +75,8 @@ def run():
         start_server(app, host=args.host, port=args.port, socket=args.socket)
     else:
         from PySide6 import QtWidgets
-        from PySide6.QtCore import QThread, QTimer
+        from PySide6.QtCore import QTimer
+
         from khoj.interface.desktop.main_window import MainWindow, ServerThread
         from khoj.interface.desktop.system_tray import create_system_tray