Update gradio_ui.py

gradio_ui.py

@@ -1,31 +1,20 @@
 import gradio as gr
 import requests
-import json
+import os
 
-# Update this if deploying elsewhere
-API_URL = "http://localhost:8000"  # For HuggingFace Spaces, use your Space URL
-
-def register(username, password):
-    try:
-        response = requests.post(
-            f"{API_URL}/register",
-            json={"username": username, "password": password}
-        )
-        return response.json().get("message", "Registration successful!")
-    except Exception as e:
-        return f"Error: {str(e)}"
+API_URL = os.getenv("API_URL", "http://localhost:8000")
 
 def login(username, password):
     try:
         response = requests.post(
             f"{API_URL}/token",
-            data={"username": username, "password": password},
+            data={"username": username, "password": password, "grant_type": "password"},
             headers={"Content-Type": "application/x-www-form-urlencoded"}
         )
         if response.status_code == 200:
             token = response.json().get("access_token")
             return f"Login successful! Token: {token[:15]}... (truncated)"
-        return "Login failed: Invalid credentials"
+        return f"Login failed: {response.json().get('detail', 'Unknown error')}"
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -35,33 +24,32 @@ def get_user_info(token):
             f"{API_URL}/users/me",
             headers={"Authorization": f"Bearer {token}"}
         )
-        return json.dumps(response.json(), indent=2)
+        return str(response.json())
     except Exception as e:
         return f"Error: {str(e)}"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Secure Authentication System")
-    
-    with gr.Tab("Register"):
-        reg_username = gr.Textbox(label="Username")
-        reg_password = gr.Textbox(label="Password", type="password")
-        reg_output = gr.Textbox(label="Output")
-        reg_button = gr.Button("Register")
-    
-    with gr.Tab("Login"):
-        login_username = gr.Textbox(label="Username")
-        login_password = gr.Textbox(label="Password", type="password")
-        login_output = gr.Textbox(label="Output")
-        login_button = gr.Button("Login")
+    gr.Markdown("""
+    # Secure JWT Authentication
+    Test credentials: username=`testuser`, password=`secret`
+    """)
     
-    with gr.Tab("User Info"):
-        token_input = gr.Textbox(label="Your JWT Token")
-        user_info_output = gr.Textbox(label="User Info", interactive=False)
-        fetch_button = gr.Button("Get User Info")
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("## Login")
+            username = gr.Textbox(label="Username")
+            password = gr.Textbox(label="Password", type="password")
+            login_btn = gr.Button("Login")
+            login_output = gr.Textbox(label="Login Result")
+        
+        with gr.Column():
+            gr.Markdown("## Get User Info")
+            token = gr.Textbox(label="Your JWT Token")
+            user_btn = gr.Button("Get User Info")
+            user_output = gr.Textbox(label="User Info")
     
-    reg_button.click(register, inputs=[reg_username, reg_password], outputs=reg_output)
-    login_button.click(login, inputs=[login_username, login_password], outputs=login_output)
-    fetch_button.click(get_user_info, inputs=token_input, outputs=user_info_output)
+    login_btn.click(login, inputs=[username, password], outputs=login_output)
+    user_btn.click(get_user_info, inputs=token, outputs=user_output)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)