Added error handling

app.py

@@ -26,9 +26,13 @@ results = get_results.main(searchquery)
 
 #st.dataframe(results)
 
-df = pd.DataFrame(results)
-
-dict_results = df.to_dict('records')
+if results:
+    df = pd.DataFrame(results)
+    dict_results = df.to_dict('records')
+else:
+    st.warning("No results found. This may be because the database is not properly configured or no matching data exists.")
+    df = pd.DataFrame()  # Empty DataFrame
+    dict_results = []
 
 st.button("Generate/View results after text-classification")
 

get_results.py

@@ -133,8 +133,14 @@ def main(shortcode):
     # Check for documents that are already present
     list_of_docs_not_present = insert_data.check_for_already_present(lst)
 
-    # Identify documents that are already present
-    list_of_docs_already_present = [docid for docid in lst.keys() if docid not in list_of_docs_not_present]
+    # Handle case where database check failed
+    if list_of_docs_not_present is None:
+        print("Error: Could not check database for existing documents. Treating all documents as new.")
+        list_of_docs_not_present = list(lst.keys())
+        list_of_docs_already_present = []
+    else:
+        # Identify documents that are already present
+        list_of_docs_already_present = [docid for docid in lst.keys() if docid not in list_of_docs_not_present]
 
     # Get text for new documents that are not already present
     lst_new_data = get_text_for_new_docs(list_of_docs_not_present, shortcode, lst)

insert_data.py

@@ -92,7 +92,11 @@ def create_task(conn, task):
     try:
         # Using a context manager to handle the cursor
         with conn.cursor() as cur:
-            cur.execute(sql_task, task)
+            try:
+                cur.execute(sql_task, task)
+            except psycopg.errors.UndefinedTable:
+                print("Warning: 'tasks' table does not exist. Skipping tasks table insertion.")
+            
             cur.execute(sql_results, task)
             conn.commit()
     except Exception as e:
@@ -115,7 +119,12 @@ def retrieve_text(conn, query):
 
     # Execute the query to fetch rows from the table
     cursor = conn.cursor()
-    cursor.execute(sql_query)
+    try:
+        cursor.execute(sql_query)
+    except psycopg.errors.UndefinedTable:
+        print("Warning: 'tasks' table does not exist. No data to retrieve.")
+        cursor.close()
+        return []  # Return empty list if table doesn't exist
 
     # Fetch all rows from the cursor
     rows = cursor.fetchall()
@@ -204,7 +213,11 @@ def add_stored_results(conn, lst):
     try:
         with conn.cursor() as cur:
             for docid in lst:
-                cur.execute(sql, (docid,))
+                try:
+                    cur.execute(sql, (docid,))
+                except psycopg.errors.UndefinedTable:
+                    print("Warning: 'tasks' table does not exist. Cannot transfer data.")
+                    return 0
             conn.commit()
             print("Stored data transferred to tasks.")
             return cur.rowcount  # Returns the total number of rows affected by the last execute call
@@ -244,10 +257,14 @@ def find_matching_text(column_value):
 def delete_sql_records(conn):
     delete_records = "DELETE FROM tasks"
 
-    cur=conn.cursor()
-    cur.execute(delete_records)
-
-    print("Deleted records")
+    cur = conn.cursor()
+    try:
+        cur.execute(delete_records)
+        print("Deleted records")
+    except psycopg.errors.UndefinedTable:
+        print("Warning: 'tasks' table does not exist. Skipping deletion.")
+    except Exception as e:
+        print(f"Error deleting records: {e}")
 
 def add_classified_results(dict_of_results, searchquery):
     conn = create_connection()