fear: add traceback

app.py

@@ -1,3 +1,4 @@
+import traceback
 from io import StringIO
 from typing import Optional
 
@@ -25,31 +26,32 @@ def process(
         text: str,
         file=None,
 ) -> (None, pd.DataFrame, str):
-    # try:
-    # load file
-    if file:
-        df = read_data(file.name)
-    elif text:
-        string_io = StringIO(text)
-        df = pd.read_csv(string_io)
-        assert len(df) >= 1, 'No input data'
-    else:
-        raise Exception('No input data')
+    try:
+        # load file
+        if file:
+            df = read_data(file.name)
+        elif text:
+            string_io = StringIO(text)
+            df = pd.read_csv(string_io)
+            assert len(df) >= 1, 'No input data'
+        else:
+            raise Exception('No input data')
 
-    # process
-    if task_name == 'Originality':
-        df = pipeline.p0_originality(df, model_name, pooling)
-    elif task_name == 'Flexibility':
-        df = pipeline.p1_flexibility(df, model_name, pooling)
-    else:
-        raise Exception('Task not supported')
+        # process
+        if task_name == 'Originality':
+            df = pipeline.p0_originality(df, model_name, pooling)
+        elif task_name == 'Flexibility':
+            df = pipeline.p1_flexibility(df, model_name, pooling)
+        else:
+            raise Exception('Task not supported')
+
+        # save
+        path = 'output.csv'
+        df.to_csv(path, index=False, encoding='utf-8-sig')
+        return None, df.iloc[:10], path
 
-    # save
-    path = 'output.csv'
-    df.to_csv(path, index=False, encoding='utf-8-sig')
-    return None, df.iloc[:10], path
-    # except Exception as e:
-    #     return {'Error': e}, None, None
+    except:
+        return {'Info': 'Something wrong', 'Error': traceback.format_exc()}, None, None
 
 
 # input