Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,8 +32,10 @@ NCBI_APIKEY = os.environ.get("NCBI_APIKEY", "")
|
|
| 32 |
# Feedback logging
|
| 33 |
FEEDBACK_PATH = os.environ.get("FEEDBACK_PATH", "feedback.csv")
|
| 34 |
PUSH_FEEDBACK = os.environ.get("PUSH_FEEDBACK", "0") == "1" # set to "1" to enable Hub upload
|
| 35 |
-
|
| 36 |
-
|
|
|
|
|
|
|
| 37 |
|
| 38 |
# Generation / toggles
|
| 39 |
ALLOW_WIKIPEDIA = False
|
|
@@ -176,13 +178,14 @@ except Exception as e:
|
|
| 176 |
_IS_IP = isinstance(index, faiss.IndexFlatIP) or "IndexFlatIP" in type(index).__name__
|
| 177 |
|
| 178 |
# ================== LOAD LLM (BASE + LORA) ==================
|
| 179 |
-
if
|
| 180 |
try:
|
| 181 |
-
login(token=
|
| 182 |
-
dlog("HF", "Login
|
| 183 |
except Exception as e:
|
| 184 |
dlog("HF", f"Login issue: {e}")
|
| 185 |
|
|
|
|
| 186 |
if ADAPTER_REPO:
|
| 187 |
ADAPTER_PATH = snapshot_download(repo_id=ADAPTER_REPO, allow_patterns=["*"])
|
| 188 |
|
|
@@ -803,21 +806,21 @@ def predict(message, chat_history, state):
|
|
| 803 |
|
| 804 |
def _push_feedback_to_hub():
|
| 805 |
"""Optional: upload feedback.csv to Space repo path analytics/feedback.csv"""
|
| 806 |
-
if not (PUSH_FEEDBACK and
|
| 807 |
return
|
| 808 |
try:
|
| 809 |
-
api = HfApi()
|
| 810 |
api.upload_file(
|
| 811 |
path_or_fileobj=FEEDBACK_PATH,
|
| 812 |
path_in_repo="analytics/feedback.csv",
|
| 813 |
repo_id=SPACE_REPO_ID,
|
| 814 |
repo_type="space",
|
| 815 |
-
token=HF_TOKEN
|
| 816 |
)
|
| 817 |
dlog("FEEDBACK", "Uploaded analytics/feedback.csv to Hub")
|
| 818 |
except Exception as e:
|
| 819 |
dlog("FEEDBACK", f"Hub upload failed: {e}")
|
| 820 |
|
|
|
|
| 821 |
def save_feedback(rating, comment, state):
|
| 822 |
if rating is None:
|
| 823 |
return "Please select a rating (1–5).", gr.update(visible=True)
|
|
|
|
| 32 |
# Feedback logging
|
| 33 |
FEEDBACK_PATH = os.environ.get("FEEDBACK_PATH", "feedback.csv")
|
| 34 |
PUSH_FEEDBACK = os.environ.get("PUSH_FEEDBACK", "0") == "1" # set to "1" to enable Hub upload
|
| 35 |
+
HF_READ_TOKEN = os.environ.get("HF_READ_TOKEN", os.environ.get("HF_TOKEN", ""))
|
| 36 |
+
HF_WRITE_TOKEN = os.environ.get("HF_WRITE_TOKEN", HF_READ_TOKEN)
|
| 37 |
+
SPACE_REPO_ID = os.environ.get("SPACE_REPO_ID", "")
|
| 38 |
+
|
| 39 |
|
| 40 |
# Generation / toggles
|
| 41 |
ALLOW_WIKIPEDIA = False
|
|
|
|
| 178 |
_IS_IP = isinstance(index, faiss.IndexFlatIP) or "IndexFlatIP" in type(index).__name__
|
| 179 |
|
| 180 |
# ================== LOAD LLM (BASE + LORA) ==================
|
| 181 |
+
if HF_READ_TOKEN:
|
| 182 |
try:
|
| 183 |
+
login(token=HF_READ_TOKEN)
|
| 184 |
+
dlog("HF", "Login (read) OK")
|
| 185 |
except Exception as e:
|
| 186 |
dlog("HF", f"Login issue: {e}")
|
| 187 |
|
| 188 |
+
|
| 189 |
if ADAPTER_REPO:
|
| 190 |
ADAPTER_PATH = snapshot_download(repo_id=ADAPTER_REPO, allow_patterns=["*"])
|
| 191 |
|
|
|
|
| 806 |
|
| 807 |
def _push_feedback_to_hub():
|
| 808 |
"""Optional: upload feedback.csv to Space repo path analytics/feedback.csv"""
|
| 809 |
+
if not (PUSH_FEEDBACK and HF_WRITE_TOKEN and SPACE_REPO_ID and os.path.exists(FEEDBACK_PATH)):
|
| 810 |
return
|
| 811 |
try:
|
| 812 |
+
api = HfApi(token=HF_WRITE_TOKEN)
|
| 813 |
api.upload_file(
|
| 814 |
path_or_fileobj=FEEDBACK_PATH,
|
| 815 |
path_in_repo="analytics/feedback.csv",
|
| 816 |
repo_id=SPACE_REPO_ID,
|
| 817 |
repo_type="space",
|
|
|
|
| 818 |
)
|
| 819 |
dlog("FEEDBACK", "Uploaded analytics/feedback.csv to Hub")
|
| 820 |
except Exception as e:
|
| 821 |
dlog("FEEDBACK", f"Hub upload failed: {e}")
|
| 822 |
|
| 823 |
+
|
| 824 |
def save_feedback(rating, comment, state):
|
| 825 |
if rating is None:
|
| 826 |
return "Please select a rating (1–5).", gr.update(visible=True)
|