Slaiwala commited on
Commit
d9a346f
·
verified ·
1 Parent(s): 9934c9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -8
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
- HF_TOKEN = os.environ.get("HF_TOKEN", "") # add in Space Secrets
36
- SPACE_REPO_ID = os.environ.get("SPACE_REPO_ID", "") # auto-provided in Spaces
 
 
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 HF_TOKEN:
180
  try:
181
- login(token=HF_TOKEN)
182
- dlog("HF", "Login successful via HF_TOKEN")
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 HF_TOKEN and SPACE_REPO_ID and os.path.exists(FEEDBACK_PATH)):
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)