Update app.py

app.py

@@ -26,7 +26,7 @@ def process_image_and_display_payload(image):
         status_message = f"<p style='color:red; font-weight:bold;'>❌ Error: {message}</p>"
         payload = {}
 
-    return status_message, payload, user, user_pass
+    return status_message, payload, user, user_pass, GROQ_API_KEY, HF_TOKEN, OPENAI_API_KEY, OPENROUTER_API_KEY
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# Gradio KeyLock Component Demo")
@@ -38,8 +38,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
     with gr.Row():
         with gr.Column():
+            gr.Markdown("## Login")
             user_name=gr.Textbox(label='User Name')
             user_pass=gr.Textbox(label='Password', type='password')
+            gr.Markdown("## API Keys")
+            user_GROQ=gr.Textbox(label='GROQ_API_KEY', type='password')
+            user_HF=gr.Textbox(label='HF_TOKEN', type='password')
+            user_OPENAI=gr.Textbox(label='OPENAI_API_KEY', type='password')
+            user_OPENROUTER=gr.Textbox(label='OPENROUTER_API_KEY', type='password')
             
         with gr.Column():
             keylock_builder = KeylockDecoderComponent(server_logic)
@@ -50,7 +56,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     image_input.upload(
         fn=process_image_and_display_payload,
         inputs=image_input,
-        outputs=[status_display, output_json, user_name, user_pass]
+        outputs=[status_display, output_json, user_name, user_pass, GROQ_API_KEY, HF_TOKEN, OPENAI_API_KEY, OPENROUTER_API_KEY]
     )
 
 demo.launch()