46e026f77f
<!-- .github/pull_request_template.md --> ## Description Add a simple GUI to add documents to Cognee and use GRAPH_COMPLETION search to get answers ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **New Features** - Introduced an interactive file search interface with intuitive controls. Users can easily upload files, enter search terms, and view results in a unified display with clear notifications during processing. - **Chores** - Updated project dependencies to include `pyside6` and `qasync` for enhanced GUI functionality. - Refined background query processing to improve the accuracy and relevance of search outcomes. - Improved code readability with formatting enhancements in the search function. <!-- end of auto-generated comment: release notes by coderabbit.ai -->
133 lines
4.4 KiB
Python
133 lines
4.4 KiB
Python
import sys
|
|
import asyncio
|
|
import cognee
|
|
|
|
try:
|
|
from PySide6.QtWidgets import (
|
|
QApplication,
|
|
QWidget,
|
|
QPushButton,
|
|
QLineEdit,
|
|
QFileDialog,
|
|
QVBoxLayout,
|
|
QLabel,
|
|
QMessageBox,
|
|
QTextEdit,
|
|
QProgressDialog,
|
|
)
|
|
from PySide6.QtCore import Qt
|
|
|
|
from qasync import QEventLoop # Import QEventLoop from qasync
|
|
except ImportError as e:
|
|
print(
|
|
"\nPlease install Cognee with optional gui dependencies or manually install missing dependencies.\n"
|
|
)
|
|
print("\nTo install with poetry use:")
|
|
print("\npoetry install -E gui\n")
|
|
print("\nOr to install with poetry and all dependencies use:")
|
|
print("\npoetry install --all-extras\n")
|
|
print("\nTo install with pip use: ")
|
|
print('\npip install ".[gui]"\n')
|
|
raise e
|
|
|
|
|
|
class FileSearchApp(QWidget):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.selected_file = None
|
|
self.init_ui()
|
|
|
|
def init_ui(self):
|
|
# Button to open file dialog
|
|
self.file_button = QPushButton("Upload File to Cognee", parent=self)
|
|
self.file_button.clicked.connect(self.open_file_dialog)
|
|
|
|
# Label to display selected file path
|
|
self.file_label = QLabel("No file selected", parent=self)
|
|
|
|
# Line edit for search input
|
|
self.search_input = QLineEdit(parent=self)
|
|
self.search_input.setPlaceholderText("Enter text to search...")
|
|
|
|
# Button to perform search; schedule the async search on click
|
|
self.search_button = QPushButton("Cognee Search", parent=self)
|
|
self.search_button.clicked.connect(lambda: asyncio.ensure_future(self._cognee_search()))
|
|
|
|
# Text output area for search results
|
|
self.result_output = QTextEdit(parent=self)
|
|
self.result_output.setReadOnly(True)
|
|
self.result_output.setPlaceholderText("Search results will appear here...")
|
|
|
|
# Progress dialog
|
|
self.progress_dialog = QProgressDialog("Processing..", None, 0, 0, parent=self)
|
|
self.progress_dialog.setWindowModality(Qt.WindowModal)
|
|
# self.progress_dialog.setAttribute(Qt.WA_DeleteOnClose)
|
|
self.progress_dialog.setCancelButton(None) # Remove the cancel button
|
|
self.progress_dialog.close()
|
|
|
|
# Layout setup
|
|
layout = QVBoxLayout()
|
|
layout.addWidget(self.file_button)
|
|
layout.addWidget(self.file_label)
|
|
layout.addWidget(self.search_input)
|
|
layout.addWidget(self.search_button)
|
|
layout.addWidget(self.result_output)
|
|
|
|
self.setLayout(layout)
|
|
self.setWindowTitle("Cognee")
|
|
self.resize(500, 300)
|
|
|
|
def open_file_dialog(self):
|
|
file_path, _ = QFileDialog.getOpenFileName(
|
|
self, "Select a File", "", "All Files (*.*);;Text Files (*.txt)"
|
|
)
|
|
if file_path:
|
|
self.selected_file = file_path
|
|
self.file_label.setText(f"Selected: {file_path}")
|
|
asyncio.ensure_future(self.process_file_async())
|
|
|
|
async def process_file_async(self):
|
|
"""Asynchronously add and process the selected file."""
|
|
# Disable the entire window
|
|
self.progress_dialog.show()
|
|
self.setEnabled(False)
|
|
try:
|
|
await cognee.add(self.selected_file)
|
|
await cognee.cognify()
|
|
except Exception as e:
|
|
QMessageBox.critical(self, "Error", f"File processing failed: {str(e)}")
|
|
# Once finished, re-enable the window
|
|
self.setEnabled(True)
|
|
self.progress_dialog.close()
|
|
|
|
async def _cognee_search(self):
|
|
"""Performs an async search and updates the result output."""
|
|
# Disable the entire window
|
|
self.setEnabled(False)
|
|
self.progress_dialog.show()
|
|
|
|
try:
|
|
search_text = self.search_input.text().strip()
|
|
result = await cognee.search(query_text=search_text)
|
|
print(result)
|
|
# Assuming result is a list-like object; adjust if necessary
|
|
self.result_output.setText(result[0])
|
|
except Exception as e:
|
|
QMessageBox.critical(self, "Error", f"Search failed: {str(e)}")
|
|
|
|
# Once finished, re-enable the window
|
|
self.setEnabled(True)
|
|
self.progress_dialog.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app = QApplication(sys.argv)
|
|
# Create a qasync event loop and set it as the current event loop
|
|
loop = QEventLoop(app)
|
|
asyncio.set_event_loop(loop)
|
|
|
|
window = FileSearchApp()
|
|
window.show()
|
|
|
|
with loop:
|
|
loop.run_forever()
|