Update utils/database.py

utils/database.py

@@ -242,6 +242,30 @@ async def generate_document_tags(content: str) -> List[str]:
     except Exception as e:
         st.error(f"Error generating tags: {e}")
         return []
+        
+def generate_document_tags(content: str) -> List[str]:
+    """Generate tags for a document using AI."""
+    try:
+        llm = ChatOpenAI(temperature=0.2, model="gpt-3.5-turbo")
+        
+        prompt = """Analyze the following document content and generate relevant tags/keywords. 
+        Focus on key themes, topics, and important terminology.
+        Return only the tags as a comma-separated list.
+        Content: {content}"""
+        
+        response = llm.invoke([
+            SystemMessage(content="You are a document analysis assistant. Generate relevant tags as a comma-separated list only."),
+            HumanMessage(content=prompt.format(content=content[:2000]))
+        ])
+        
+        # Extract content from the AI message
+        tags_text = response.content
+        # Split the comma-separated string into a list
+        tags = [tag.strip() for tag in tags_text.split(',')]
+        return tags
+    except Exception as e:
+        st.error(f"Error generating tags: {e}")
+        return []
 
 def add_document_tags(conn: sqlite3.Connection, document_id: int, tags: List[str]) -> bool:
     """Add tags to a document."""