Update stock_vector_db.py

stock_vector_db.py

@@ -74,3 +74,36 @@ class HFVectorDB:
             st.write("⚠️ No index found, returning empty results")
             return []
         return self.index.similarity_search(query, k=k)
+
+
+
+def save_top_picks_json(top_picks, date, path="top_picks.jsonl"):
+    record = {
+        "date": date.isoformat(),
+        "top_picks": top_picks
+    }
+    with open(path, "a") as f:
+        f.write(json.dumps(record) + "\n")
+    st.write(f"✅ Saved top picks to {path}")
+
+def add_top_picks_to_vector_db(vector_db, top_picks, date):
+    docs = []
+    for pick in top_picks:
+        content = (
+            f"{pick['company']} ({pick['ticker']}):\n"
+            f"Sentiment: {pick['sentiment']}\n"
+            f"Critical News: {pick['critical_news']}\n"
+            f"Impact: {pick['impact_summary']}\n"
+            f"Action: {pick['action']}\n"
+            f"Reason: {pick['reason']}"
+        )
+        metadata = {
+            "ticker": pick["ticker"],
+            "company": pick["company"],
+            "sentiment": pick["sentiment"],
+            "action": pick["action"],
+            "date": date.isoformat()
+        }
+        docs.append(Document(page_content=content, metadata=metadata))
+    vector_db.add_documents(docs)
+    st.write("✅ Added top picks to vector DB")