Create app.py

app.py

@@ -0,0 +1,84 @@
+# app.py
+import gradio as gr
+import requests
+import os
+
+# 从Secrets获取API地址。URL现在指向新的端点 /predict_df
+API_URL = os.getenv("API_URL", "http://YOUR_SERVER_IP:8000/predict_df")
+
+# 固定的model_name，也可以做成下拉框让用户选择
+MODEL_NAME = "/data/tsq/MODELS/Qwen2.5-1.5B-Instruct"
+
+def call_ml_server(csv_text_input, final_result_only_checkbox):
+    """
+    调用后端API的函数
+    """
+    if "YOUR_SERVER_IP" in API_URL:
+        raise gr.Error("错误：API服务器地址未配置！请在Space的Settings -> Secrets中设置API_URL。")
+    
+    # 1. 检查输入是否为空
+    if not csv_text_input.strip():
+        return None # 如果输入为空，返回None，Gradio会清空输出
+        
+    # 2. 构建发送到API的JSON payload
+    payload = {
+        "csv_data": csv_text_input,
+        "model_name": MODEL_NAME,
+        "only_final_result": final_result_only_checkbox
+    }
+    
+    try:
+        # 3. 发送POST请求
+        response = requests.post(API_URL, json=payload, timeout=60) # 延长超时
+        response.raise_for_status()
+        
+        # 4. 直接返回JSON结果，Gradio的Dataframe组件会自动处理
+        return response.json()
+        
+    except requests.exceptions.HTTPError as e:
+        # 如果API返回错误（如500），尝试解析错误详情
+        error_detail = e.response.json().get('detail', e.response.text)
+        raise gr.Error(f"API返回错误: {error_detail}")
+    except requests.exceptions.RequestException as e:
+        raise gr.Error(f"请求API时出错: {e}")
+    except Exception as e:
+        raise gr.Error(f"发生未知错误: {e}")
+
+# 示例数据
+example_data = "id,subject,relation,object,prompt_source\n001,France,capital city of,Paris,she traveled to France for the first time\n002,Japan,capital city of,Tokyo,he is planning a trip to Japan"
+
+# 创建Gradio界面
+with gr.Blocks() as iface:
+    gr.Markdown("## 云服务器DataFrame推理前端")
+    gr.Markdown("在下方文本框中输入CSV格式的数据，然后点击“提交”。数据将被发送到私有云服务器进行处理。")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            csv_input = gr.Textbox(
+                lines=10, 
+                label="输入CSV数据", 
+                placeholder="请在这里粘贴你的CSV数据...",
+                value=example_data # 设置默认值
+            )
+            final_result_checkbox = gr.Checkbox(label="Only Final Result", value=False)
+            submit_btn = gr.Button("提交", variant="primary")
+            
+        with gr.Column(scale=3):
+            output_df = gr.Dataframe(label="服务器返回结果", wrap=True)
+    
+    gr.Examples(
+        examples=[
+            [example_data, False],
+            ["id,subject,relation,object,prompt_source\n003,Germany,capital city of,Berlin,Germany is a country in Europe", True]
+        ],
+        inputs=[csv_input, final_result_checkbox]
+    )
+    
+    submit_btn.click(
+        fn=call_ml_server,
+        inputs=[csv_input, final_result_checkbox],
+        outputs=output_df
+    )
+
+# 启动界面
+iface.launch()