Update app.py

app.py

@@ -513,10 +513,12 @@ logging.basicConfig(level=logging.INFO, filename='translation.log', filemode='a'
                             format='%(asctime)s - %(levelname)s - %(message)s')
 
 # Main function to handle the translation workflow
-def main(dataset_url, model_type, output_dataset_name, range_specification):
+def main(dataset_url, model_type, output_dataset_name, range_specification, token: gr.OAuthToken | None, profile: gr.OAuthProfile | None):
     try:
         # Login to Hugging Face
-        token = login()
+        if token is None or profile is None or token.token is None or profile.username is None:
+            return "### You must be logged in to use this service."
+        
         if token:
             logging.info("Logged in to Hugging Face")
 
@@ -616,7 +618,7 @@ with gr.Blocks(theme=theme) as demo:
             output = gr.Textbox(label="Output", lines=1)
 
     submit_btn = gr.Button("Translate Dataset", variant="primary")
-    submit_btn.click(main, inputs=[dataset_url, model_type, output_dataset_name, range_specification, token_input], outputs=output)
+    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.")