deepcode-hku 1.0.1__py3-none-any.whl

tools/pdf_utils.py

@@ -0,0 +1,52 @@
+"""
+PDF utility functions for the DeepCode agent system.
+"""
+
+from pathlib import Path
+import PyPDF2
+
+
+def read_pdf_metadata(file_path: Path) -> dict:
+    """Read PDF metadata with proper encoding handling."""
+    try:
+        print(f"\nAttempting to read PDF metadata from: {file_path}")
+        with open(file_path, "rb") as file:
+            pdf_reader = PyPDF2.PdfReader(file)
+            info = pdf_reader.metadata
+            first_page = pdf_reader.pages[0]
+            text = first_page.extract_text()
+            lines = text.split("\n")[:10]
+
+            title = None
+            authors = []
+
+            if info:
+                title = info.get("/Title", "").strip().replace("\x00", "")
+                author = info.get("/Author", "").strip().replace("\x00", "")
+                if author:
+                    authors = [author]
+
+            if not title and lines:
+                title = lines[0].strip()
+
+            if not authors and len(lines) > 1:
+                for line in lines[1:3]:
+                    if "author" in line.lower() or "by" in line.lower():
+                        authors = [line.strip()]
+                        break
+
+            return {
+                "title": title if title else "Unknown Title",
+                "authors": authors if authors else ["Unknown Author"],
+                "year": info.get("/CreationDate", "")[:4] if info else "Unknown Year",
+                "first_lines": lines,
+            }
+
+    except Exception as e:
+        print(f"\nError reading PDF: {str(e)}")
+        return {
+            "title": "Error reading PDF",
+            "authors": ["Unknown"],
+            "year": "Unknown",
+            "first_lines": [],
+        }

ui/__init__.py

@@ -0,0 +1,43 @@
+"""
+UI Module
+
+Streamlit application user interface components module
+
+Contains the following submodules:
+- styles: CSS styles
+- components: UI components
+- layout: Page layout
+- handlers: Event handlers
+- streamlit_app: Main application
+- app: Application entry
+"""
+
+__version__ = "1.0.0"
+__author__ = "DeepCode Team"
+
+# Import main components
+from .layout import main_layout
+from .components import display_header, display_features, display_status
+from .handlers import initialize_session_state
+from .styles import get_main_styles
+
+# Import application main function
+try:
+    from .streamlit_app import main as streamlit_main
+except ImportError:
+    # Fallback to absolute import if relative import fails
+    import sys
+    import os
+
+    sys.path.insert(0, os.path.dirname(__file__))
+    from streamlit_app import main as streamlit_main
+
+__all__ = [
+    "main_layout",
+    "display_header",
+    "display_features",
+    "display_status",
+    "initialize_session_state",
+    "get_main_styles",
+    "streamlit_main",
+]

ui/app.py

@@ -0,0 +1,13 @@
+"""
+DeepCode UI Application Entry Point
+
+This file serves as the unified entry point for the UI module
+"""
+
+from .streamlit_app import main
+
+# Directly export main function for external calls
+__all__ = ["main"]
+
+if __name__ == "__main__":
+    main()