google collab jupyter_dash bug fix

Author: connectaman

novalad/api/client.py

@@ -7,7 +7,7 @@
     get_file_extension, get_filename,is_cloud_storage_path,
     is_valid_url, extract_filename_from_url
 )
-from novalad.api.exception import APIError, InvalidArgumentException, FileFormatNotSupportedException
+from novalad.api.exception import APIError, InvalidArgumentException, FileFormatNotSupportedException, FileNotUploaded
 from novalad.utils.progress import tqdm
 
 class NovaladClient(BaseAPIClient):
@@ -79,12 +79,14 @@ def upload(self, file_path: Optional[str] = None, folder_path: Optional[str] = N
         :raises InvalidArgumentException: If both or neither of file_path and folder_path are provided.
         :raises APIError: If the upload process encounters an error.
         """
+        self.uploaded = False
         if (file_path is None and folder_path is None) or (file_path is not None and folder_path is not None):
             raise InvalidArgumentException("You must provide either 'file_path' or 'folder_path', but not both.")
 
         if file_path and is_filepath(file_path):
             upload_url = self._upload_url(file_path)
             self._upload_to_cloud(file_path, upload_url)
+            self.uploaded = True
 
         elif folder_path and is_folderpath(folder_path):
             files = get_files_from_folder(folder_path)
@@ -95,6 +97,7 @@ def upload(self, file_path: Optional[str] = None, folder_path: Optional[str] = N
             for file in tqdm(supported_files, desc="Uploading files"):
                 upload_url = self._upload_url(file)
                 self._upload_to_cloud(file, upload_url)
+            self.uploaded = True
 
     def run(self, url : str = None,
             skip_non_important_images : bool = True,
@@ -108,6 +111,9 @@ def run(self, url : str = None,
             "skip_header_footer" : skip_header_footer
         }
 
+        if not self.uploaded:
+            raise FileNotUploaded("File not Uploaded, Either the path/url is invalid or document is corrupted.")
+
         if (self.file_id is None and url is None) or (self.file_id is not None and url is not None):
             raise InvalidArgumentException("You must upload local file or provide 'url', but not both.")
 

novalad/api/exception.py

@@ -11,6 +11,9 @@ class InvalidArgumentException(Exception):
     pass
 
 
+class FileNotUploaded(Exception):
+    pass
+
 class FileFormatNotSupportedException(Exception):
     pass
 

novalad/render/graph.py

@@ -1,40 +1,34 @@
-import os
 import sys
 import json
-
-# Universal imports
 import dash_cytoscape as cyto
 from dash import html, dcc
 from dash.dependencies import Input, Output
 
 # Use JupyterDash in Colab, Dash locally
-try:
-    if "google.colab" in sys.modules:
-        from jupyter_dash import JupyterDash as DashApp
-    else:
-        from dash import Dash as DashApp
-except ImportError:
-    from dash import Dash as DashApp
+if "google.colab" in sys.modules:
+    from jupyter_dash import JupyterDash
+    DashApp = JupyterDash
+else:
+    from dash import Dash
+    DashApp = Dash
 
 
-def render_knowledge_graph(data: dict, save: bool = False, filename: str = "kg.html") -> None:
+def render_knowledge_graph(data: dict) -> None:
     """
     Render an interactive knowledge graph using Dash and Dash Cytoscape.
 
     Args:
         data (dict): Knowledge graph data with nodes and edges.
-        save (bool): If True, saves the graph as a static HTML file.
-        filename (str): Output filename for HTML export.
 
     Returns:
         None
     """
-    data = data["data"].get("knowledge_graphs", {})
+    graph_data = data["data"].get("knowledge_graphs", {})
 
-    # Convert data to Cytoscape elements
+    # Convert to Cytoscape elements
     elements = []
 
-    for node in data["nodes"]:
+    for node in graph_data["nodes"]:
         elements.append({
             "data": {
                 "id": node["id"],
@@ -47,7 +41,7 @@ def render_knowledge_graph(data: dict, save: bool = False, filename: str = "kg.h
             }
         })
 
-    for edge in data["edges"]:
+    for edge in graph_data["edges"]:
         elements.append({
             "data": {
                 "source": edge["fromId"],
@@ -56,8 +50,8 @@ def render_knowledge_graph(data: dict, save: bool = False, filename: str = "kg.h
             }
         })
 
+    # Build the app
     app = DashApp(__name__)
-
     app.layout = html.Div([
         html.H3("Interactive Knowledge Graph"),
         html.Div(id="node-info", style={"margin": "10px", "font-size": "14px", "color": "#F58634"}),
@@ -98,12 +92,8 @@ def display_click_info(node_data):
             return f"Selected Node: {node_data['hover_text']}"
         return "Click on a node to see details"
 
-    if save:
-        app.run_server(mode="inline", debug=False)
-        # Use selenium or dash-export to save actual HTML content if needed
-        print("Note: Static export not implemented in this version.")
+    # Show app inline if in Colab, else open in browser
+    if "google.colab" in sys.modules:
+        app.run_server(mode='inline', debug=True)
     else:
-        if "google.colab" in sys.modules:
-            app.run_server(mode="inline", debug=True)
-        else:
-            app.run_server(debug=True)
+        app.run_server(debug=True)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "novalad"
-version = "0.1.4"
+version = "0.1.5"
 description = "Novalad: AI-powered platform for transforming unstructured documents like PDFs and PowerPoints into machine-readable, structured data."
 authors = ["Novalad <info@novalad.ai>"]
 license = "Apache-2.0"