Update app.py

app.py

@@ -3,9 +3,10 @@ import gradio as gr
 import xgboost as xgb
 import numpy as np
 import pandas as pd
+import requests
 from huggingface_hub import hf_hub_download
 
-# **📌 先运行 `preprocess.py` 进行数据处理**
+# **📌 先运行 `preprocess.py` 处理数据**
 if not os.path.exists("processed_data.csv"):
     print("📌 运行 `preprocess.py` 进行数据处理...")
     os.system("python preprocess.py")
@@ -22,6 +23,24 @@ model_path = hf_hub_download(repo_id="YDluffy/lottery_prediction", filename="lot
 model = xgb.XGBRegressor()
 model.load_model(model_path)
 
+# **📌 调用 LLM API 解析用户输入**
+LLM_API_URL = "https://your-llm-space.gradio.app"  # 替换为 LLM 服务器地址
+
+def get_prediction_from_llm(user_input):
+    # 调用 LLM 提取预测所需的特征
+    response = requests.post(LLM_API_URL + "/api/predict", json={"input": user_input})
+    llm_output = response.json().get("prediction", "")
+
+    # 解析 LLM 输出（示例解析）
+    year, period = 2025, 16
+    nums = [5, 12, 23, 34, 45, 56]
+    special = 7
+
+    # **📌 预测**
+    prediction = predict_lottery(year, period, *nums, special)
+    
+    return f"预测的号码是: {prediction}\n\n模型解析的特征：{llm_output}"
+
 # **📌 预测函数**
 def predict_lottery(year, period, num1, num2, num3, num4, num5, num6, special):
     # 从历史数据查找对应的月份和日期
@@ -44,16 +63,11 @@ def predict_lottery(year, period, num1, num2, num3, num4, num5, num6, special):
 
 # **📌 Gradio Web 界面**
 iface = gr.Interface(
-    fn=predict_lottery,
-    inputs=[
-        gr.Number(label="年份"), gr.Number(label="期数"),
-        gr.Number(label="号码1"), gr.Number(label="号码2"), gr.Number(label="号码3"),
-        gr.Number(label="号码4"), gr.Number(label="号码5"), gr.Number(label="号码6"),
-        gr.Number(label="特别号码")
-    ],
+    fn=get_prediction_from_llm,
+    inputs=gr.Textbox(label="请输入问题或期号信息"),
     outputs="text",
     title="六合彩预测模型",
-    description="输入期号和历史开奖号码，预测指定期开奖的号码"
+    description="通过与 LLM 交互，解析输入信息并进行预测"
 )
 
 # **📌 启动 Gradio 应用**