Update app.py

app.py

@@ -1,19 +1,20 @@
 from fastapi import FastAPI, UploadFile, File
 from typing import List
-import chromadb
+from sentence_transformers import SentenceTransformer
+from transformers import pipeline
+import asyncio
 
+# --- FastAPI app ---
 app = FastAPI(title="Eroha AgentAPI", version="2.1.1")
 
-client = chromadb.Client()
-collection = client.get_or_create_collection("eroha_docs")
-
+# --- Lazy initialization ---
 embedder = None
 summarizer = None
 
 @app.get("/")
 def home():
     return {
-        "message": "Eroha AgentAPI is alive!",
+        "message": "✅ Eroha AgentAPI is alive!",
         "routes": ["/check", "/summarize"]
     }
 
@@ -28,17 +29,33 @@ def check_health():
 @app.post("/summarize")
 async def summarize_text(files: List[UploadFile] = File(...)):
     global embedder, summarizer
+
+    # Lazy load models (only on first call)
     if embedder is None or summarizer is None:
-        from sentence_transformers import SentenceTransformer
-        from transformers import pipeline
         embedder = SentenceTransformer("all-MiniLM-L6-v2")
-        summarizer = pipeline("text2text-generation", model="google/flan-t5-base")
+        summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
+    # Collect text from uploaded files
     texts = []
     for file in files:
         content = await file.read()
-        texts.append(content.decode("utf-8"))
+        texts.append(content.decode("utf-8", errors="ignore"))
 
+    # Combine into one large string
     full_text = "\n".join(texts)
-    summary = summarizer(full_text, max_length=150, min_length=40, do_sample=False)
-    return {"summary": summary[0]["generated_text"]}
+
+    # Create embedding (optional — for debugging)
+    embedding = embedder.encode(full_text[:512])  # just preview embedding
+
+    # Summarize
+    summary = summarizer(full_text, max_length=200, min_length=50, do_sample=False)
+
+    return {
+        "summary": summary[0]["summary_text"],
+        "embedding_preview": embedding[:5].tolist() if hasattr(embedding, "tolist") else [],
+    }
+
+# --- Health endpoint ---
+@app.get("/ping")
+def ping():
+    return {"status": "running"}