add authentication for gradio

app.py

@@ -5,8 +5,13 @@ from llm_api.exceptions import RefusalError
 # from llm_api.langchain_api import extract_info, follow_structure
 from llm_api.native_api import extract_info, follow_structure
 from openai import BadRequestError
+import os
 
 
+# Load environment variables
+attr_user = os.getenv('ATTR_USER')
+attr_pass = os.getenv('ATTR_PASS')
+
 load_dotenv('.env', override=True)
 garment_categories = ['Auto', 'Upper Garment', 'Lower Garment', 'Hat', 'Luggage', 'Handbag', 'Makeup', 'Chair', 'Carpet', 'Watch', 'Cookware', 'Tableware', 'Sleepware', 'BathroomItem']
 schema_categories = ['Auto', 'UpperGarmentProperties', 'LowerGarmentProperties', 'HatProperties', 'LuggageProperties', 'HandbagProperties', 'MakeupProperties', 'ChairProperties', 'CarpetProperties', 'WatchProperties', 'CookwareProperties', 'TablewareProperties', 'SleepwareProperties', 'BathroomItemProperties']
@@ -113,9 +118,11 @@ with gr.Blocks(title='Internal Demo for Attribution') as demo:
         structured_schema = gr.TextArea(label='Structured Schema', placeholder='Schema', visible=True, value=schema_structured_text, interactive=True)
     
     button_extract.click(extract_free_fn, [gallery, category,provider, free_schema], [status_free_text, free_attributes, probs_text])
-    button_follow.click(follow_schema_fn, [free_attributes, category, provider, probs_text, structured_schema], [status_structured_text, schema_attributes])
+    button_follow.click(follow_schema_fn, [free_attributes, provider, probs_text, structured_schema], [status_structured_text, schema_attributes])
+
+auth = (attr_user, attr_pass)
 
 # demo.launch(server_name="0.0.0.0", server_port=7688, share=False, ssl_verify=True, ssl_certfile=None, ssl_keyfile=None, auth=None)
 # demo.launch(server_name="0.0.0.0", server_port=7860, share=False, ssl_verify=False, ssl_certfile="server.crt", ssl_keyfile="server.key", auth=None)
-demo.launch()
+demo.launch(auth=auth)