Update main.py

main.py

@@ -230,6 +230,66 @@ async def chat(request: ChatRequest, background_tasks: BackgroundTasks, api_key:
         logger.error(f"Error in chat endpoint: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Error in chat endpoint: {str(e)}")
 
+from typing import Dict, List
+import os
+import json
+
+class IndexMetadata(BaseModel):
+    index_id: str
+    document_count: int
+    created_at: str = Field(default_factory=lambda: datetime.now().isoformat())
+    size_bytes: int
+
+@app.get("/list_indexes/", response_model=Dict[str, List[IndexMetadata]], tags=["Index Operations"])
+async def list_indexes():
+    """
+    List all available indexes and their metadata.
+    
+    Returns a dictionary containing:
+    - List of indexes with their metadata (document count, creation date, size)
+    """
+    try:
+        indexes_path = "/app/indexes"
+        if not os.path.exists(indexes_path):
+            return {"indexes": []}
+        
+        indexes = []
+        for index_id in os.listdir(indexes_path):
+            index_path = os.path.join(indexes_path, index_id)
+            if os.path.isdir(index_path):
+                try:
+                    # Get document count from document_list.json
+                    doc_list_path = os.path.join(index_path, "document_list.json")
+                    with open(doc_list_path, "r") as f:
+                        documents = json.load(f)
+                        doc_count = len(documents)
+                    
+                    # Calculate total size of the index
+                    total_size = 0
+                    for root, dirs, files in os.walk(index_path):
+                        total_size += sum(os.path.getsize(os.path.join(root, file)) for file in files)
+                    
+                    # Get creation time of the index directory
+                    created_at = datetime.fromtimestamp(os.path.getctime(index_path)).isoformat()
+                    
+                    indexes.append(IndexMetadata(
+                        index_id=index_id,
+                        document_count=doc_count,
+                        created_at=created_at,
+                        size_bytes=total_size
+                    ))
+                    
+                except Exception as e:
+                    logger.error(f"Error processing index {index_id}: {str(e)}")
+                    continue
+        
+        logger.info(f"Successfully retrieved metadata for {len(indexes)} indexes")
+        return {"indexes": indexes}
+        
+    except Exception as e:
+        logger.error(f"Error listing indexes: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error listing indexes: {str(e)}")
+
 
 
 @app.on_event("startup")