Create gradio_ui.py

gradio_ui.py

@@ -0,0 +1,67 @@
+import gradio as gr
+import requests
+import json
+
+# 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)}"
+
+def login(username, password):
+    try:
+        response = requests.post(
+            f"{API_URL}/token",
+            data={"username": username, "password": 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"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+def get_user_info(token):
+    try:
+        response = requests.get(
+            f"{API_URL}/users/me",
+            headers={"Authorization": f"Bearer {token}"}
+        )
+        return json.dumps(response.json(), indent=2)
+    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")
+    
+    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")
+    
+    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)
+
+if __name__ == "__main__":
+    demo.launch()