Update app.py

app.py

@@ -3,8 +3,6 @@ import requests
 import json
 
 def send_post_request(url, const_smiles, var_smiles, main_cls, minor_cls, delta_value, target_name, num):
-    # 构造POST请求的JSON数据
-    # url:https://songyou-llm-fastapi.hf.space/generate
     data = {
         "constSmiles": const_smiles,
         "varSmiles": var_smiles,
@@ -16,39 +14,55 @@ def send_post_request(url, const_smiles, var_smiles, main_cls, minor_cls, delta_
     }
 
     try:
-        # 发送POST请求
         response = requests.post(url, headers={"Content-Type": "application/json"}, data=json.dumps(data))
-        print("------response-----")
-        print(f"'status_code': {response.status_code}")
-        print(f"'text': {response.text}")
-        print(response.json())
-        # 返回请求响应内容
+        return response.json()
+    except Exception as e:
+        return {"error": str(e)}
+
+def send_get_request(url, smiles):
+    params = {"smiles": smiles}
+    try:
+        response = requests.get(url, params=params)
         return response.json()
     except Exception as e:
         return {"error": str(e)}
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# POST Request Sender")
-
-    with gr.Row():
-        url = gr.Textbox(label="Request URL")
-        const_smiles = gr.Textbox(label="Constant SMILES")
-        var_smiles = gr.Textbox(label="Variable SMILES")
-        main_cls = gr.Textbox(label="Main Class")
-        minor_cls = gr.Textbox(label="Minor Class")
-        delta_value = gr.Textbox(label="Delta Value")
-        target_name = gr.Textbox(label="Target Name")
-        num = gr.Number(label="Number", precision=0)
-
-    output = gr.Textbox(label="Response")
-
-    submit_btn = gr.Button("Submit")
-    submit_btn.click(
-        send_post_request, 
-        inputs=[url, const_smiles, var_smiles, main_cls, minor_cls, delta_value, target_name, num], 
-        outputs=output
-    )
-
-# 启动 Gradio 界面
+    gr.Markdown("# API Request Sender")
+
+    with gr.Tab("POST /generate"):
+        gr.Markdown("Send a POST request to `/generate` endpoint")
+        with gr.Row():
+            post_url = gr.Textbox(label="Request URL", value="https://songyou-llm-fastapi.hf.space/generate")
+            const_smiles = gr.Textbox(label="Constant SMILES")
+            var_smiles = gr.Textbox(label="Variable SMILES")
+            main_cls = gr.Textbox(label="Main Class")
+            minor_cls = gr.Textbox(label="Minor Class")
+            delta_value = gr.Textbox(label="Delta Value")
+            target_name = gr.Textbox(label="Target Name")
+            num = gr.Number(label="Number", precision=0)
+
+        post_output = gr.Textbox(label="Response")
+        post_submit_btn = gr.Button("Submit POST Request")
+        post_submit_btn.click(
+            send_post_request, 
+            inputs=[post_url, const_smiles, var_smiles, main_cls, minor_cls, delta_value, target_name, num], 
+            outputs=post_output
+        )
+
+    with gr.Tab("GET /fragmentize"):
+        gr.Markdown("Send a GET request to `/fragmentize` endpoint")
+        with gr.Row():
+            get_url = gr.Textbox(label="Request URL", value="https://songyou-llm-fastapi.hf.space/fragmentize")
+            smiles_input = gr.Textbox(label="SMILES String")
+
+        get_output = gr.Textbox(label="Response")
+        get_submit_btn = gr.Button("Submit GET Request")
+        get_submit_btn.click(
+            send_get_request,
+            inputs=[get_url, smiles_input],
+            outputs=get_output
+        )
+
 demo.launch()