vectordb geminiapi

使用Gemini API，MongoDB矢量搜索和簡化UI的電影搜索

描述：

該項目實施了一個電影搜索應用程序，該應用程序利用了：

• Google Gemini API：為有效的矢量嵌入生成，捕獲了電影圖的語義含義。
• MongoDB矢量搜索：檢索與用戶搜索查詢的語義相似的圖片。
• 簡化：用於創建一個用戶友好和交互式Web應用程序。

特徵：

• 用戶可以輸入電影搜索查詢。
• 該應用程序使用Gemini API生成查詢的向量表示。
• 然後，它在MongoDB中執行矢量搜索，以查找具有相似情節嵌入的電影。
• 用戶界面顯示檢索到的電影的詳細信息，包括標題，繪圖摘要（如果可用），以及可選的海報（如果數據中存在URL）。

要求：

• Python 3.x
• 必需的庫：
  • google-generativeai （用於雙子API）
  • pymongo （用於MongoDB相互作用）
  • streamlit （用於Web應用程序開發）

安裝：

1. 克隆這個存儲庫。
2. 創建虛擬環境（推薦）：

   python -m venv env
   source env/bin/activate  # Windows: envScriptsactivate.bat

3. 安裝所需的庫：

   pip install google-generativeai pymongo streamlit

設定:

1. 配置Gemini API：
   • 獲取Google Cloud項目並啟用Gemini API。
   • 創建一個API鍵並相應地設置環境變量GOOGLE_API_KEY 。
2. 連接到MongoDB：
   • 設置一個包含電影數據的集合的MongoDB數據庫。該集合應包括帶有title ， plot （用於矢量搜索）的字段的文檔，以及可選的poster （用於圖像顯示）。
   • 用MongoDB連接字符串和數據庫/集合名稱替換connection_string.py中的佔位符。

用法：

1. 運行應用程序：

   streamlit run movie_recs.py

2. 在文本輸入字段中輸入搜索查詢，然後按“ Enter”。
3. 該應用程序將顯示一部電影列表，這些電影與您的查詢以及它們的詳細信息匹配。