chat_settings.py

# -*- coding: utf-8 -*-
"""
Created on Mon Mar  4 13:41:35 2019

@author: Danish
"""

"""
ChatSettings class
"""
import copy
from googletrans import Translator
t = Translator()

class ChatSettings(object):
    """Contains settings for a chat session.
    """
    def __init__(self, model_hparams, inference_hparams):
        """
        Args:
            inference_hparams: the loaded InferenceHparams instance to use as default for this chat session
        """
        self.show_question_context = False
        self.show_all_beams = False
        self.enable_auto_punctuation = True
        self.model_hparams = None
        self.inference_hparams = None
        
        self._default_model_hparams = model_hparams
        self._default_inference_hparams = inference_hparams
        self.reset_to_defaults()
    
    def reset_to_defaults(self):
        """Reset all settings to defaults
        """
        self.show_question_context = False
        self.show_all_beams = False
        self.enable_auto_punctuation = True
        self.model_hparams = copy.copy(self._default_model_hparams)
        self.inference_hparams = copy.copy(self._default_inference_hparams)

    def To_query(query):
        
        question = t.translate([query], dest='en')
        for get_question in question:
            t_question=get_question.text
        return t_question   

    def To_answer(reply):

        answer = t.translate([reply], dest='ur')
        for get_answer in answer:
            t_answer=get_answer.text
        return t_answer