Upload app.py

app.py

@@ -18,6 +18,14 @@ def company_scoring(company_name):
     scoring_result_text = ''
     for score in scoring_result:
         scoring_result_text = scoring_result_text + str(score) + ','
+
+    total = 0
+    for i in range(0, 5):
+        total += scoring_result[i]
+    for i in range(6, 10):
+        total += scoring_result[i]
+    total = str(total)
+
     # url_list = ', '.join(urls)
     if group == 'Group 1-1':
         header = header1_1
@@ -33,12 +41,10 @@ def company_scoring(company_name):
         header = header5
 
     header = ', '.join(header)
-    if isinstance(related_url_list, list):
-        related_url_list = ','.join(related_url_list)
     if group == 'Group 1-2' or group == 'Group 5':
         result = company_name + ',' + group + ',' + related_url_list
     else:
-        result = company_name + ',' + group + ',' + scoring_result_text + related_url_list
+        result = company_name + ',' + group + ',' + scoring_result_text + total + ',' + related_url_list
     output = header + '\n' + result
     return output, None, None, None, None, None, None, None
 
@@ -69,7 +75,7 @@ def company_list_scoring(file):
 
 
 def answer(input, file):
-    if file is not None:
+    if input == '':
         return company_list_scoring(file)
     else:
         return company_scoring(input)
@@ -77,7 +83,7 @@ def answer(input, file):
 
 demo = gr.Interface(fn=answer,
                     inputs=["textbox",
-                            gr.File(label="input file")],
+                            gr.File(label="input.csv")],
                     outputs=["textbox",
                              gr.File(label="Classification_result.csv"),
                              gr.File(label="Group1-1_result.csv"),