Avoid errors due to empty tables

monarch-initiative · Aug 14, 2024 · adafb1d · adafb1d
1 parent b700904
commit adafb1d
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 13 deletions.
diff --git a/src/curate_gpt/agents/mapping_agent.py b/src/curate_gpt/agents/mapping_agent.py
@@ -138,20 +138,24 @@ def match(
         response = model.prompt(prompt)
         response_text = response.text()
         mappings = []
-        for m in json.loads(response_text):
-            m = str(m)
-            if m in objects:
-                object_id = objects.get(m)["id"]
-            else:
-                object_id = m
-            mappings.append(
-                Mapping(
-                    subject_id=query,
-                    object_id=object_id,
-                    predicate=MappingPredicate.SAME_AS,
-                    prompt=prompt,
+        try:
+            for m in json.loads(response_text):
+                m = str(m)
+                if m in objects:
+                    object_id = objects.get(m)["id"]
+                else:
+                    object_id = m
+                mappings.append(
+                    Mapping(
+                        subject_id=query,
+                        object_id=object_id,
+                        predicate=MappingPredicate.SAME_AS,
+                        prompt=prompt,
+                    )
                 )
-            )
+        except json.decoder.JSONDecodeError:
+            return MappingSet(mappings=mappings, prompt=prompt, response_text=response_text)
+
         return MappingSet(mappings=mappings, prompt=prompt, response_text=response_text)
 
     def categorize_mappings(

diff --git a/src/curate_gpt/app/app.py b/src/curate_gpt/app/app.py
@@ -190,6 +190,8 @@ def ask_chatbot(query, expand=False) -> ChatResponse:
 
 
 def html_table(rows: List[dict]) -> str:
+    if len(rows) == 0:
+        rows = [{"No data": "No data"}]
     hdr = rows[0].keys()
     html_content = '<table border="1">'
     cols = [f"<th>{h}</th>" for h in hdr]