Big Small Query

app.py

@@ -12,16 +12,19 @@ def check_query(query):
         t_text = split[1]
 
         if t_text.startswith("### Tables"):
+            query_params = dict()
             tables = t_text.split('\n')
-            print(tables[1:])
-            return('big')
+            query_params['tables'] = tables[1:]
+            query_params['q'] = q_text.split('\n')[1]
+            print(query_params)
+            return query_params
         else:
-            return('error')
-    return('small')
+            return 'error'
+    return 'small'
 
 def answer_question(query):
     query_check = check_query(query)
-    if query_check == 'big':
+    if len(query_check) > 1:
         return("BIG TABLE")
     if query_check == 'small':
         return('SMALL TABLE')