Minor clean-up

functions.py

@@ -6,7 +6,7 @@ import time
 from pyzotero import zotero
 from paperqa import Docs
 from lxml import html
-from models import Icons, Message, Messages
+from models import Icons, Message
 
 
 def reset_answer():

models.py

@@ -1,66 +1,67 @@
-import gradio as gr
 from enum import Enum
 
+
 class Icons(Enum):
-  def __str__(self):
-      return str(self.value)
-  DOC = "📄"
-  ERR = "❌"
-  INDEX = "🗄️"
-  INFO = "ℹ️"
-  OK = "👌"
-  SUCCESS = "✅"
-  WAIT = "⌛"
-  WARN = "⚠️"
+    def __str__(self):
+        return str(self.value)
+    DOC = "📄"
+    ERR = "❌"
+    INDEX = "🗄️"
+    INFO = "ℹ️"
+    OK = "👌"
+    SUCCESS = "✅"
+    WAIT = "⌛"
+    WARN = "⚠️"
+
 
 class Message():
-  def __init__(self, icon, content):
-    self.icon = icon
-    self.content = content
-  
-  def __str__(self):
-    return f"{self.icon} {self.content}"
+    def __init__(self, icon, content):
+        self.icon = icon
+        self.content = content
+
+    def __str__(self):
+        return f"{self.icon} {self.content}"
+
 
 class Messages():
-  def __init__(self, messages=[]):
-    self.messages = messages
+    def __init__(self, messages=[]):
+        self.messages = messages
 
-  def __str__(self):
-    return f"""
+    def __str__(self):
+        return f"""
     <div class="messages" style="padding: 1rem; background-color: #fffbe7; font-size: 0.8rem;">
       {("").join([f"<div>{x}</div>" for x in self.messages])}
     </div>
     """
-  
-  def append(self, new_message):
-    self.messages.append(new_message)
 
-  def set(self, messages):
-    self.messages = messages
+    def append(self, new_message):
+        self.messages.append(new_message)
+
+    def set(self, messages):
+        self.messages = messages
 
 # class Message():
 
 #   def standing_by(self):
 #     return "<div>👌 Standing by...</div>"
-    
+
 #   def not_ready(self):
 #     return """
 #     <div style="padding: 1rem; background-color: #fffbe7; font-size: 0.8rem;">
 #       You have to select a Zotero collection to proceed.
 #     </div>
 #     """
-  
+
 #   def openai_api_key(self):
 #     return """
 #     <div style="padding: 1rem; background-color: #fcd7da; font-size: 0.8rem;">
 #       OpenAI API key is either missing or incorrect.
 #     </div>
 #     """
-  
+
 #   def use_queries(queries):
 #     query_str = ", ".join([f"<span style="font-weight: bold;">{q}</span>" for q in queries])
 #     return f"<div>Search your Zotero collection for {query_str}"
- 
 
 
 # def update_status(messages):
@@ -68,4 +69,4 @@ class Messages():
 #   <div class="messages" style="padding: 1rem; background-color: #fffbe7; font-size: 0.8rem;">
 #     {("").join(messages)}
 #   </div>
-#   """)
+#   """)