app modified

app.py

@@ -201,7 +201,7 @@ class Model:
              
         
         return  cat, porcentaje      
-   def predict(self):
+   def predict(self,etiquetas):
         
         categoria, porcentaje = self.categorizar_texto(self.texto)
         print(categoria, porcentaje)
@@ -226,19 +226,24 @@ class Model:
             
            
             out1 = self.salida_json(tokens,predicted_tokens_classes) #spanish solo palabras sensibles
-                        
-            out2 = self.salida_texto_anonimizado(ids,self.reemplazo_fake(predicted_tokens_classes)) #español texto completo    
+
+            if etiquetas:            
+                out2 = self.salida_texto_anonimizado(ids,predicted_tokens_classes) #solo identificadores
+            else:    
+                out2 = self.salida_texto_anonimizado(ids,self.reemplazo_fake(predicted_tokens_classes)) #español texto completo    
             
         else:
             
             new_tokens,ig_tokens=self.reordenacion_tokens(tokens)
             new_identificadores = self.reordenacion_identificadores(ig_tokens,predicted_tokens_classes)
            
-            out1 = self.salida_json(new_tokens,new_identificadores),
-           
-            
+              
             
-            out2 = self.salida_texto(new_tokens,self.reemplazo_fake(new_identificadores))
+            out1 = self.salida_json(new_tokens,new_identificadores),
+            if etiquetas: 
+                out2 = self.salida_texto(new_tokens,new_identificadores)#solo identificadores
+            else:
+                out2 = self.salida_texto(new_tokens,self.reemplazo_fake(new_identificadores))
             
            
         return (
@@ -406,10 +411,12 @@ model = Model()
 def get_model():
         return model
 
-def procesar(texto,archivo):
+def procesar(texto,archivo, etiquetas):
+    
+    
     if len(texto)>0: 
         model.identificacion_idioma(texto)
-        return model.predict(),gr.Dataframe(),gr.File()
+        return model.predict(etiquetas),gr.Dataframe(),gr.File()
     else:
 
         if archivo.name.split(".")[1]=="csv":
@@ -446,7 +453,7 @@ def procesar(texto,archivo):
                 #return "", df, df.to_csv(sep='\t', encoding='utf-8',index=False) 
                 return "", df_new, df_new.to_csv(sep='\t', encoding='utf-8',index=False)
             
-demo = gr.Interface(fn=procesar,inputs=["text",gr.File()] , outputs=["text",gr.Dataframe(interactive=False),"text"])
+demo = gr.Interface(fn=procesar,inputs=["text",gr.File(), "checkbox"] , outputs=["text",gr.Dataframe(interactive=False),"text"])
        #                                                      
 demo.launch(share=True)