Update main.py

main.py

@@ -100,8 +100,8 @@ async def main():
             """
         )
         
-        with gr.Tab("文章の類似度計算"):
-            gr.Markdown("## 2つの文章を入力して類似度を計算します")
+        with gr.Tab("コンテキスト理解"):
+            gr.Markdown("## 2つの文章を入力して類似度を計算します（コンテキスト理解機能）")
             with gr.Row():
                 text_input1 = gr.Textbox(label="文章1", lines=3, placeholder="例: 今日の天気は晴れです。")
                 text_input2 = gr.Textbox(label="文章2", lines=3, placeholder="例: 今日は良い天気ですね。")
@@ -118,6 +118,9 @@ async def main():
             generate_button = gr.Button("ベクトルを生成", variant="primary")
             embeddings_output = gr.JSON(label="生成された埋め込みベクトル")
             generate_button.click(fn=get_embeddings, inputs=texts_input, outputs=embeddings_output)
+        
+        # カスタムAPIエンドポイントの追加: 2つのテキストの類似度計算を非同期で公開
+        gr.api(calculate_similarity, api_name="calculate_similarity")
 
     # --- 6. Gradioアプリの起動（高性能化: キュー拡張と同時実行制限増加） ---
     demo.queue(max_size=500, default_concurrency_limit=20)  # キューサイズ拡大と同時実行制限増加