Update gradio_app.py

gradio_app.py

@@ -56,6 +56,7 @@ with open('system.prompt', 'r', encoding='utf-8') as f:
     prompt = f.read()
 
 def generate_answer(request: str, max_tokens: int = 256, language: str = "en", custom_prompt: str = None):
+    print("Request:", request, "\nMax tokens:", max_tokens, "\nLanguage:", language, "\nCustom prompt:", custom_prompt, "\n")
     try:
         maxTokens = max_tokens if 16 <= max_tokens <= 256 else 64
         if isinstance(custom_prompt, str):
@@ -69,6 +70,7 @@ def generate_answer(request: str, max_tokens: int = 256, language: str = "en", c
         output = llm(userPrompt, max_tokens=maxTokens, stop=["User:"], echo=False)
         text = output["choices"][0]["text"]
         if language in languages:
+            print("Translating from en to", language)
             encoded_input = translator_tokenizer(text, return_tensors="pt")
             generated_tokens = translator_model.generate(
                 **encoded_input, forced_bos_token_id=translator_tokenizer.get_lang_id(language)
@@ -76,11 +78,14 @@ def generate_answer(request: str, max_tokens: int = 256, language: str = "en", c
             translated_text = translator_tokenizer.batch_decode(
                 generated_tokens, skip_special_tokens=True
             )[0]
+            print("Translated:", translated_text, "\nOriginal:", text)
             return translated_text
+        print(text)
         return text
     except Exception as e:
         print(e)
         return "Oops! Internal server error. Check the logs of space/instance."
+    print("\n\n\n")
 
 print("! LOAD GRADIO INTERFACE !")
 demo = gr.Interface(