Update app.py

app.py

@@ -704,7 +704,7 @@ if __name__ == "__main__":
                     with gr.Column():
                         tokenize_output = gr.Code(label="Result (JSON)", language="json")
                 
-                tokenize_btn.click(tokenize_text, [tokenize_input], [tokenize_output])
+                tokenize_btn.click(tokenize_text, [tokenize_input], [tokenize_output], api_name="/predict")
                 logger.info("[INIT] Tokenize tab components configured")
             
             with gr.Tab("💾 Prompt Cache"):
@@ -722,8 +722,8 @@ if __name__ == "__main__":
                         cache_get_btn = gr.Button("Retrieve", variant="secondary")
                         cache_get_output = gr.Code(label="Result", language="json")
                 
-                cache_store_btn.click(cache_prompt, [cache_key_input, cache_value_input], [cache_store_output])
-                cache_get_btn.click(get_cached_prompt, [cache_get_input], [cache_get_output])
+                cache_store_btn.click(cache_prompt, [cache_key_input, cache_value_input], [cache_store_output], api_name="/predict_2")
+                cache_get_btn.click(get_cached_prompt, [cache_get_input], [cache_get_output], api_name="/predict_3")
                 logger.info("[INIT] Prompt Cache tab components configured")
             
             with gr.Tab("⚡ Response Cache"):
@@ -741,8 +741,8 @@ if __name__ == "__main__":
                         resp_get_btn = gr.Button("Get Response", variant="secondary")
                         resp_get_output = gr.Code(label="Result", language="json")
                 
-                resp_cache_btn.click(cache_response, [resp_hash_input, resp_value_input], [resp_cache_output])
-                resp_get_btn.click(get_cached_response, [resp_get_input], [resp_get_output])
+                resp_cache_btn.click(cache_response, [resp_hash_input, resp_value_input], [resp_cache_output], api_name="/predict_4")
+                resp_get_btn.click(get_cached_response, [resp_get_input], [resp_get_output], api_name="/predict_5")
                 logger.info("[INIT] Response Cache tab components configured")
             
             with gr.Tab("💰 Token Accounting"):
@@ -755,7 +755,7 @@ if __name__ == "__main__":
                 calc_btn = gr.Button("Calculate Cost", variant="primary")
                 calc_output = gr.Code(label="Result (JSON)", language="json")
                 
-                calc_btn.click(calculate_token_cost, [username_input, duration_input], [calc_output])
+                calc_btn.click(calculate_token_cost, [username_input, duration_input], [calc_output], api_name="/predict_6")
                 logger.info("[INIT] Token Accounting tab components configured")
             
             with gr.Tab("📊 Stats"):
@@ -764,7 +764,7 @@ if __name__ == "__main__":
                 stats_btn = gr.Button("Get Stats", variant="primary")
                 stats_output = gr.Code(label="Statistics (JSON)", language="json")
                 
-                stats_btn.click(get_cache_stats, None, [stats_output])
+                stats_btn.click(get_cache_stats, None, [stats_output], api_name="/predict_7")
                 logger.info("[INIT] Stats tab components configured")
             
             with gr.Tab("🏥 Health"):
@@ -773,7 +773,7 @@ if __name__ == "__main__":
                 health_btn = gr.Button("Check Health", variant="primary")
                 health_output = gr.Code(label="Health Status (JSON)", language="json")
                 
-                health_btn.click(get_backend_health, None, [health_output])
+                health_btn.click(get_backend_health, None, [health_output], api_name="/predict_8")
                 logger.info("[INIT] Health tab components configured")
             
             logger.info("[INIT] All tabs created successfully")