Upload folder using huggingface_hub

db.py

@@ -129,7 +129,7 @@ class SanatanDatabase:
             }
             Or a dict with "error" key if something went wrong.
         """
-        logger.info("fetching document #%d from [%s]", index, collection_name)
+        logger.info("fetching %s #%d from [%s]", unit_name, index, collection_name)
         collection = self.chroma_client.get_or_create_collection(name=collection_name)
 
         try:
@@ -160,6 +160,7 @@ class SanatanDatabase:
                 result.update(metadatas[0])
             return result
         else:
+            print("No data available")
             return {"error": "No data available."}
 
 

server.py

@@ -214,7 +214,7 @@ async def get_scripture(req: ScriptureRequest):
     raw_doc = SanatanDatabase().fetch_document_by_index(
         collection_name=config["collection_name"],
         index=req.unit_index,
-        unit_name=config.get("unit_field", config.get("unit"))
+        unit_name=config.get("unit_field", config.get("unit")),
     )
 
     if not raw_doc or isinstance(raw_doc, str):
@@ -239,16 +239,15 @@ async def get_scripture(req: ScriptureRequest):
 async def get_scripture_configs():
     scriptures = []
     for s in SanatanConfig().scriptures:
-        num_units = SanatanDatabase().count(
-            collection_name=s["collection_name"]
-        )
-
+        num_units = SanatanDatabase().count(collection_name=s["collection_name"])
         scriptures.append(
             {
                 "name": s["name"],  # e.g. "bhagavad_gita"
                 "title": s["title"],  # e.g. "Bhagavad Gita"
                 "unit": s["unit"],  # e.g. "verse" or "page"
-                "total" : num_units
+                "total": num_units,
+                "enabled": s["name"]
+                in ("divya_prabandham", "chathusloki", "vishnu_sahasranamam"),
             }
         )
-    return {"scriptures": scriptures}
+    return {"scriptures": sorted(scriptures, key=lambda s: s["title"])}