Update app.py

app.py

@@ -24,24 +24,23 @@ def question_answer(context, question):
   return prediction
 
 def split(text):
-    # words = word_tokenize(text)
-    # context, question = '', ''
-    # act = False
+    context, question = '', ''
+    act = False
+    tmp = ''
+    for t in text:
+        tmp += t
+        if len(tmp) == 4:
+            tmp = tmp[1:]
+            if tmp == '///':
+                act = True
 
-    # for w in words:
-    #     if w == '///':
-    #         act = True
-            
-    #     if act == False:
-    #         context += w + ' '
-    #     else:
-    #         if w == '///':
-    #             w = ''
-    #         question += w + ' '
+        if act == True:
+            question += t
 
-    # context = context[:-1]
-    # question = question[1:-1]
-    return text, text
+        if act == False:
+            context += t
+        
+    return context[:-2], question[1:]
     
 # def greet(texts):
 #     context, question = split(texts)
@@ -50,7 +49,7 @@ def split(text):
 def greet(text):
     context, question = split(text)
     # answer = question_answer(context, question)
-    return word_tokenize(text)
+    return question
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()