Raspberry pi codes

.gitignore

@@ -9,3 +9,4 @@ Gemma2_2B/.cache
 **/*/wandb
 Gemma2_2B/outputs/
 Gemma2_2B/gemma-2-2b-it-therapist
+**/*/__pycache__

Audio.py

@@ -0,0 +1,10 @@
+import pyaudio
+
+audio = pyaudio.PyAudio()
+
+# List all audio devices
+for i in range(audio.get_device_count()):
+    info = audio.get_device_info_by_index(i)
+    print(f"Device Index {i}: {info['name']} - {info['hostApi']}")
+
+audio.terminate()

Gemma2_2B/gemma-2-2b-therapy.modelfile

@@ -0,0 +1,16 @@
+
+FROM /home/TADBot/TADBot/Gemma2_2B/gemma-2-2b-it-therapist/gemma-2-2b-it-therapist-Q4_K_M.gguf
+
+TEMPLATE """{{- range $i, $_ := .Messages }}
+{{- $last := eq (len (slice $.Messages $i)) 1 }}
+{{- if or (eq .Role "user") (eq .Role "system") }}<start_of_turn>user
+{{ .Content }}<end_of_turn>
+{{ if $last }}<start_of_turn>model
+{{ end }}
+{{- else if eq .Role "assistant" }}<start_of_turn>model
+{{ .Content }}{{ if not $last }}<end_of_turn>
+{{ end }}
+{{- end }}
+{{- end }}"""
+PARAMETER stop <start_of_turn>
+PARAMETER stop <end_of_turn>

Output.py

@@ -0,0 +1,10 @@
+import pyttsx3
+
+engine = pyttsx3.init('espeak')
+voices = engine.getProperty('voices')
+print("Available voices:")
+for voice in voices:
+    print(f"- {voice.id}")
+engine.setProperty('voice', voices[14].id)  # Use the first available voice
+engine.say("Hello from Raspberry Pi")
+engine.runAndWait()

TAD.py

@@ -1,4 +1,5 @@
-import ollama 
+from ollama import Client
+import ollama
 import chromadb
 import speech_recognition as sr
 import requests 
@@ -32,7 +33,7 @@ message_history = [
 ]
 convo = []
 
-modelname = "ms"
+llm = Client(host='http://localhost:11434')
 
 
 def create_vector_db(conversations):
@@ -58,12 +59,12 @@ def create_vector_db(conversations):
 def stream_response(prompt):
     
     convo.append({'role': "user", 'content': prompt})
-    output = ollama.chat(model = modelname, messages = convo)
+    output = llm.chat(model = "TADBot", messages = convo)
     response = output['message']['content']
 
     print("TADBot: ") 
     print(response)
-    engine = pyttsx3.init()
+    engine = pyttsx3.init('espeak')
     engine.say(response)
     engine.runAndWait()
 

pyproject.toml

@@ -5,51 +5,20 @@ description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
-    "ruff>=0.7.3",
-    "torch==2.5.1+cu124",
-    "torchvision==0.20.1+cu124",
-    "torchaudio==2.5.1+cu124",
-    "huggingface-hub>=0.26.2",
-    "timm>=1.0.11",
-    "thop>=0.1.1.post2209072238",
-    "deepface==0.0.93",
-    "bitsandbytes>=0.44.1",
-    "peft>=0.13.2",
-    "trl>=0.12.1",
-    "ninja==1.11.1.1",
-    "numpy>=1.26.4",
-    "transformers>=4.46.2",
-    "datasets>=3.1.0",
-    "tensorboard>=2.18.0",
-    "jsonlines>=4.0.0",
-    "python-dotenv>=1.0.1",
-    "ipykernel>=6.29.5",
-    "ipywidgets>=8.1.5",
-    "pyyaml>=6.0.2",
-    "torch-tb-profiler>=0.4.3",
-    "tensorflow>=2.18.0",
-    "wandb>=0.18.7",
-]
-
-[tool.uv.sources]
-torch = { index = "pytorch" }
-torchvision = { index = "pytorch" }
-torchaudio = { index = "pytorch" }
-deepface = {git = "https://github.com/serengil/deepface", tag="v0.0.93"}
-
-[[tool.uv.index]]
-name = "pytorch"
-url = "https://download.pytorch.org/whl/cu124"
-explicit = true
-
-[dependency-groups]
-dev = [
-    "deepface>=0.0.93",
-    "numpy<2",
-    "opencv-python>=4.10.0.84",
-    "thop>=0.1.1.post2209072238",
-    "timm>=1.0.11",
     "torch>=2.5.1",
     "torchaudio>=2.5.1",
     "torchvision>=0.20.1",
+    "deepface>=0.0.93",
+    "timm>=1.0.11",
+    "thop>=0.1.1.post2209072238",
+    "scikit-learn>=1.5.2",
+    "matplotlib>=3.9.2",
+    "torchsampler>=0.1.2",
+    "tf-keras>=2.18.0",
+    "pyaudio>=0.2.14",
+    "pyttsx3>=2.98",
+    "ollama>=0.3.3",
+    "chromadb>=0.5.20",
+    "speechrecognition>=3.11.0",
 ]
+deepface = {git = "https://github.com/serengil/deepface", tag="v0.0.93"}