Update app.py

app.py

@@ -509,8 +509,14 @@ def translate_dataset(train_url, local_parquet_path, input_file_path, output_fil
         logging.error(f"Failed to upload {output_file_path} to Hugging Face: {e}")  
 
 # Setup logging configuration
-logging.basicConfig(level=logging.INFO, filename='translation.log', filemode='a',
-                            format='%(asctime)s - %(levelname)s - %(message)s')
+log_stream = io.StringIO()
+logging.basicConfig(level=logging.INFO,
+                    format='%(asctime)s - %(levelname)s - %(message)s',
+                    handlers=[
+                        logging.FileHandler("translation.log", mode='a'),
+                        logging.StreamHandler(log_stream)
+                    ])
+logger = logging.getLogger(__name__)
 
 # Main function to handle the translation workflow
 def main(dataset_url, model_type, output_dataset_name, range_specification, token: gr.OAuthToken | None, profile: gr.OAuthProfile | None):
@@ -570,15 +576,15 @@ def main(dataset_url, model_type, output_dataset_name, range_specification, toke
             return "Login failed. Please try again."
     except Exception as e:
         logging.error(f"An error occurred in the main function: {e}")
-        return f"An error occurred: {e}"
+        return f"An error occurred: {e}\n\n### Logs:\n{log_stream.getvalue()}"
 
 # Gradio interface setup
 gradio_title = "🧐 WMT21 Dataset Translation"
 gradio_desc = """This tool translates datasets using the WMT21 translation model.
 ## 💭 What Does This Tool Do:
 - Translates datasets based on the selected model type.
-- Uploads the translated dataset to Hugging Face.
-## 📊 Dataset Types:
+- Uploads the translated dataset to Hugging Face."""
+datasets_desc = """## 📊 Dataset Types:
 - **mix**: 
   - `prompt`: List of dictionaries with 'content' and 'role' fields (multi-turn conversation).
   - `chosen`: Single dictionary with 'content' and 'role' fields.
@@ -592,8 +598,8 @@ gradio_desc = """This tool translates datasets using the WMT21 translation model
   - `chosen`: List of dictionaries with 'content' and 'role' fields.
   - `rejected`: List of dictionaries with 'content' and 'role' fields.
 ## 🛠️ Backend:
-The translation backend runs on the Hugging Face Hub API.
-"""
+The translation backend runs on the Hugging Face Hub API."""
+
 # Define the theme
 theme = gr.themes.Soft(text_size="lg", spacing_size="lg")
 
@@ -620,6 +626,6 @@ with gr.Blocks(theme=theme) as demo:
     submit_btn = gr.Button("Translate Dataset", variant="primary")
     submit_btn.click(main, inputs=[dataset_url, model_type, output_dataset_name, range_specification], outputs=output)
 
-    gr.Markdown(value="We sincerely thank our community members for their extraordinary contributions to this project.")
+    gr.Markdown(datasets_desc)
 
 demo.launch(share=True)