Update gradio_app.py

gradio_app.py

@@ -56,7 +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")
+    logs = f"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):
@@ -64,13 +64,13 @@ def generate_answer(request: str, max_tokens: int = 256, language: str = "en", c
         else:
             userPrompt = prompt + "\n\nUser: " + request + "\nAssistant: "
     except:
-        return "Not enough data! Check that you passed all needed data."
+        return "Not enough data! Check that you passed all needed data.", logs
     
     try:
         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)
+            logs += f"\nTranslating 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)
@@ -78,13 +78,13 @@ 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
+            logs += f"\nTranslated: {translated_text}\nOriginal: {text}"
+            return translated_text, logs
+        logs += f"\nOriginal: {text}"
+        return text, logs
     except Exception as e:
         print(e)
-        return "Oops! Internal server error. Check the logs of space/instance."
+        return "Oops! Internal server error. Check the logs of space/instance.", logs
     print("\n\n\n")
 
 print("! LOAD GRADIO INTERFACE !")
@@ -96,7 +96,10 @@ demo = gr.Interface(
         gr.components.Dropdown(label="Target Language", value="en", choices=["en"]+languages),
         gr.components.Textbox(label="Custom system prompt"),
     ],
-    outputs=["text"],
+    outputs=[
+        gr.components.Textbox(label="Output"),
+        gr.components.Textbox(label="Logs")
+    ],
     title=title,
     description=desc,
     allow_flagging='never'