Spaces:
Runtime error
Runtime error
Upload 4 files
Browse files- utils/__init__.py +1 -0
- utils/image_processor.py +1 -0
- utils/rag_pipeline.py +17 -0
- utils/tts_handler.py +7 -0
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
|