diff --git a/README.md b/README.md index 48471a3..d8b7b7d 100644 --- a/README.md +++ b/README.md @@ -185,3 +185,5 @@ Distributed under the MIT License. See [`LICENSE`](LICENSE) for more information

Made with 💙 by Samay

+ +# Test direct push diff --git a/ai_engine/local_pipeline.py b/ai_engine/local_pipeline.py index a966690..8d685bb 100644 --- a/ai_engine/local_pipeline.py +++ b/ai_engine/local_pipeline.py @@ -7,7 +7,7 @@ from typing import TypedDict, List, Dict, Any from faster_whisper import WhisperModel -from langchain_ollama import ChatOllama +from langchain_groq import ChatGroq from langchain_core.messages import SystemMessage, HumanMessage from langgraph.graph import StateGraph, START, END @@ -305,10 +305,10 @@ def node_transcribe(state: VideoState) -> VideoState: return {"audio_path": audio, "video_duration": dur, "words": words, "segments": sentences, "status": "Transcribed"} def node_plan(state: VideoState) -> VideoState: - print("\n--- [Ollama] Strategizing Top 0.1% Cinematic B-Roll Placements ---") + print("\n--- [Groq] Strategizing Top 0.1% Cinematic B-Roll Placements ---") segments = state["segments"] - llm = ChatOllama(model="qwen3:4b", temperature=0.2) + llm = ChatGroq(model="llama3-70b-8192", temperature=0.2, groq_api_key=os.getenv("GROQ_API_KEY")) transcript_summary = "" for i, seg in enumerate(segments): diff --git a/ai_engine/requirements.txt b/ai_engine/requirements.txt index 8615f9a..5c10188 100644 --- a/ai_engine/requirements.txt +++ b/ai_engine/requirements.txt @@ -15,3 +15,8 @@ google-generativeai==0.8.2 # YouTube Download yt-dlp>=2024.1.0 +# AI Pipeline (Cloud-ready) +faster_whisper>=1.0.3 +langgraph>=0.1.5 +langchain_groq>=0.1.6 +langchain_core>=0.2.11