AptlyDigital commited on
Commit
20d7f8e
·
verified ·
1 Parent(s): 09e432b

Upload 4 files

Browse files
utils/__init__.py ADDED
@@ -0,0 +1 @@
 
 
1
+ # Empty init files
utils/image_processor.py ADDED
@@ -0,0 +1 @@
 
 
1
+
utils/rag_pipeline.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from models.embedder import Embedder
2
+ from documents.chunker import DocumentChunker
3
+
4
+ class RAGPipeline:
5
+ def __init__(self):
6
+ self.embedder = Embedder()
7
+ self.chunker = DocumentChunker()
8
+
9
+ def add_documents(self, text):
10
+ chunks = self.chunker.chunk_text(text)
11
+ embeddings = self.embedder.embed(chunks)
12
+ for i, (chunk, embedding) in enumerate(zip(chunks, embeddings)):
13
+ self.embedder.collection.add(
14
+ documents=[chunk],
15
+ embeddings=[embedding],
16
+ ids=[str(i)]
17
+ )
utils/tts_handler.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ # Google TTS handler - will be implemented with API key
2
+ class TTSHandler:
3
+ def __init__(self):
4
+ pass
5
+
6
+ def text_to_speech(self, text):
7
+ return "audio_path.mp3" # Placeholder