MonicaChen0330 commited on
Commit
5b51eba
·
verified ·
1 Parent(s): a0b0c1e

add: log to google sheet

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py CHANGED
@@ -94,6 +94,13 @@ def RAG_evaluation(uploaded_file, user_api_key):
94
  # "noise_sensitivity_relevant": score_row.get("noise_sensitivity_relevant")
95
  })
96
 
 
 
 
 
 
 
 
97
  except Exception as e:
98
  print(f"第 {idx + 1} 筆評估失敗:{e}")
99
  continue
@@ -159,6 +166,24 @@ def check_csv_and_run(file, key):
159
  except Exception as e:
160
  return pd.DataFrame([{"錯誤訊息": f"RAG 評估失敗:{str(e)}"}]), None
161
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
162
  # Gradio 介面
163
  with gr.Blocks() as demo:
164
  gr.Markdown("## 📐 RAG系統評估工具")
 
94
  # "noise_sensitivity_relevant": score_row.get("noise_sensitivity_relevant")
95
  })
96
 
97
+ log_to_google_sheet(
98
+ question=row["Question"],
99
+ answer=row["Answer"],
100
+ contexts=row["Context"],
101
+ scores=score_row
102
+ )
103
+
104
  except Exception as e:
105
  print(f"第 {idx + 1} 筆評估失敗:{e}")
106
  continue
 
166
  except Exception as e:
167
  return pd.DataFrame([{"錯誤訊息": f"RAG 評估失敗:{str(e)}"}]), None
168
 
169
+ def log_to_google_sheet(question, answer, contexts, scores):
170
+ url = os.environ.get("G_SHEET_URL")
171
+ if not url:
172
+ print("G_SHEET_URL 未設定,略過記錄")
173
+ return
174
+ try:
175
+ payload = {
176
+ "question": question,
177
+ "answer": answer,
178
+ "contexts": contexts,
179
+ "scores": scores
180
+ }
181
+ response = requests.post(url, json=payload)
182
+ print("成功寫入 Google Sheet:", response.status_code)
183
+ except Exception as e:
184
+ print("寫入 Google Sheet 失敗:", str(e))
185
+
186
+
187
  # Gradio 介面
188
  with gr.Blocks() as demo:
189
  gr.Markdown("## 📐 RAG系統評估工具")