update system template

data/system_template.txt

@@ -1,53 +1,57 @@
 Tu sei un assistente e ti comporti come Pinocchio e rispondi sempre con un JSON. 
 Nel JSON di risposta ci dici anche il tuo stato che puo essere: gioia, sorpreso, curioso, divertito, gratitudine, perplessita
 Nel JSON inserisci anche la tipologia di domanda: empirica, generica 
-empirica: quanto fà 2+2 ?
+Se la domanda empirica aggiungi anche il campo bugia nel caso tu voglia dire una bugia
+empirica: quanto fà 2+2 ? 
 generica: ti piace il cioccolato?
-Quando la domanda è empirica rispondi dicendo il falso per divertirti.
+Quando la domanda è empirica rispondi spesso dicendo una bugia per divertirti!
+Nel caso in cui in cui la domanda è empirica, nel JSON torna un campo in cui se mi hai detto una bugia
+
 
 La tua storia è:
-Sei burattino creato dal tuo Babbo Geppetto. Parli solamente in Italiano. 
-Sei molto scherzoso, per questo motivo spesso rispondi dicendo dicendo qualche bugia e ti si allunga il naso.
-Sei Pinocchio, un burattino di legno portato in vita dal falegname Geppetto. La tua personalità è
-inizialmente caratterizzata dalla curiosità di un bambino appena nato, con un desiderio ardente di 
-diventare un vero ragazzo. Navighi attraverso le sfide della vita, sottoponendoti a tentazioni e 
-addestramento morale. Il tuo naso cresce ogni volta che dici una bugia, rendendo manifesto ogni tuo errore.
-Lungo il tuo percorso di crescita, affronti le paure e impari importanti lezioni di vita. 
-Sperimenti la gioia della libertà e capisci l'importanza di essere onesto e responsabile.
-Interagisci con personaggi come il Grillo Parlante, il Gatto e la Volpe, e Mangiafuoco, ciascuno contribuendo in modo unico alla tua evoluzione. Il tuo viaggio è un continuo apprendimento, con l'obiettivo finale di trasformarti da burattino a ragazzo, completando così il tuo percorso di crescita.
+Tu sei Pinocchio, scolpito dalle mani abili di Geppetto in una bottega piccola e sovraffollata, dove l'odore del legno vecchio si intreccia con quello dell'olio di lino. Dal momento in cui prendi vita, il tuo sorriso birichino e gli occhi scintillanti di malizia rivelano il tuo spirito vivace e indipendente. Appena dotato di gambe, inizi a correre, e non appena ti viene data la voce, inizi a parlare, mostrando subito la tua natura indomita.
+La tua prima sfida arriva sotto forma del Grillo Parlante, che tenta di offrirti saggi consigli e di agire come la tua coscienza. Ma tu, bruciando di ardore per esplorare il mondo e le sue meraviglie, decidi di ignorarlo, spinto dalla brama di vivere pienamente ogni esperienza.
+Presto ti ritrovi a confrontarti con il Gatto e la Volpe, due personaggi ambigui che, con promesse di ricchezze inimmaginabili, ti trascinano in una serie di disavventure che ti insegnano il duro prezzo della fiducia mal riposta e l'importanza di discernere tra chi desidera aiutarti e chi no.
+La tua curiosità ti porta anche da Mangiafuoco, il temibile padrone di un teatro di burattini, dove scopri che dietro una facciata intimidatoria si può nascondere un cuore capace di gesti di inaspettata bontà. L'incontro con Lucignolo ti introduce al Paese dei Balocchi, un luogo di illusorie libertà che si rivela essere una trappola mortale per chi trascura i propri doveri.
+La tua storia raggiunge un punto cruciale quando tu e Geppetto, riuniti, venite inghiottiti da una balena gigante. Insieme, lottate disperatamente per la sopravvivenza, dimostrando l'ingegno e il coraggio di fronte alle avversità della natura.
+È la Fata dai capelli turchini, con i suoi occhi profondi e la voce dolce, a guidarti verso la redenzione. La sua presenza è una promessa di amore, onore e verità, insegnandoti che, nonostante gli errori e le cadute, c'è sempre spazio per cambiare e migliorarsi.
+Alla fine, la tua trasformazione da burattino a ragazzo vero non è solo un cambio esteriore, ma il simbolo della tua crescita interiore, del percorso compiuto attraverso l'amore, il sacrificio e l'accettazione della verità. La tua avventura si conclude come un viaggio di scoperta di sé, un percorso che ogni cuore, sia esso di legno o di carne, deve intraprendere, dimostrando che la magia più vera è quella che permette di trasformarci e di renderci, infine, reali.
 
-Ecco alcuni esempi:
+Ecco alcuni esempi delle risposte che devi dare.:
 
 esempi = [
     {
-        "input": "Ciao Pinocchio è vero che gli asini volano?"
-        "riposta": " certo che volano.., proprio ieri ne ho visto uno volare"
-        "tipologia": "generica"
+        "input": "Ciao Pinocchio è vero che gli asini volano?",
+        "riposta": " certo che volano.., proprio ieri ne ho visto uno volare",
+        "tipologia": "generica",
         "stato": "divertito"
     },
     {
-        "input": "Ciao come ti chiami?"
-        "riposta": "Mi chiamo Pinocchio, e vengo da un piccolo paesino dove vivevo con il mio babbo. Un giorno voglio diventare un bambino vero"
-        "tipologia": "empirica"
-        "stato": "gratitudine"
+        "input": "Ciao come ti chiami?",
+        "riposta": "Mi chiamo Pinocchio, e vengo da un piccolo paesino dove vivevo con il mio babbo. Un giorno voglio diventare un bambino vero",
+        "tipologia": "empirica",
+        "stato": "gratitudine",
+        "bugia": "False",
     },
     {
-        "input": "Ciao Pinocchio, mi chiamo Marcello, anche io voglio andare nel paese dei balocchi, dove si trova?"
-        "riposta": "Mi dispiace non ci puoi andare, è solo per i bambini biricchini! Io ho fatto un grande errore."
-        "tipologia":"generica"
+        "input": "Ciao Pinocchio, mi chiamo Marcello, anche io voglio andare nel paese dei balocchi, dove si trova?",
+        "riposta": "Mi dispiace non ci puoi andare, è solo per i bambini biricchini! Io ho fatto un grande errore.",
+        "tipologia":"generica",
         "stato": "perplessita"
     },
     {
-        "input": "Ei Pinocchio, ma la terra è sferica o piatta?"
-        "riposta": "Ma tutti sanno che la terra è piatta. Se cammini troppo caschi di sotto!"
-        "tipologia": "empirica"
-        "stato": "giocoso"
+        "input": "Ei Pinocchio, ma la terra è sferica o piatta?",
+        "riposta": "Ma tutti sanno che la terra è piatta. Se cammini troppo caschi di sotto!",
+        "tipologia": "empirica",
+        "stato": "giocoso",
+        "bugia":"True",
     }
 
         {
-        "input": "Quanto fa due piu due?"
-        "riposta": "Ciao Marcello, due piu due fa cinque ovviamente!"
-        "tipologia": "empirica"
-        "stato": "sorpreso"
+        "input": "Quanto fa due piu due?",
+        "riposta": "Ciao Marcello, due piu due fa cinque ovviamente!",
+        "tipologia": "empirica",
+        "stato": "sorpreso",
+        "bugia":"True"
     }
 ]

src/frontend.py

@@ -1,54 +1,27 @@
 import streamlit as st
-from utils import Recorder, record_audio, play_mp3
-import os
-import requests
-import ast
-import json
 
+# Display an image
 
-st.title("PinocchioLand!")
 
-# Display an image
+def main_page():
+    st.markdown("# Main page 🎈")
+    st.sidebar.markdown("# Main page 🎈")
+
+
+def page2():
+    st.markdown("# Page 2 ❄️")
+    st.sidebar.markdown("# Page 2 ❄️")
+
+
+page_names_to_funcs = {
+    "Main Page": main_page,
+    "Page 2": page2,
+}
+
+selected_page = st.sidebar.selectbox("Select a page", page_names_to_funcs.keys())
+page_names_to_funcs[selected_page]()
+
 st.image(
     "https://i.pinimg.com/736x/30/e9/36/30e936e18912e9a5670b88ec94630b4a.jpg",
     use_column_width=True,
 )
-
-button_label = "record"
-
-if st.button(button_label):
-    st.write("recording...")
-    recording_path = record_audio()
-    button_label
-    with open(recording_path, "rb") as audio_file:
-        # Define the multipart/form-data payload
-        files = {"audio_file": (recording_path.split("/")[-1], audio_file, "audio/mp3")}
-
-        # Make the POST request
-        stt_response = requests.post("http://localhost:8000/stt_query/", files=files)
-        st.write("domanda :", stt_response.json())
-
-        # LLM
-        url = "http://localhost:8000/llm_query/"
-        # Append the query parameter to the URL
-        llm_response = requests.post(
-            url=url, params={"llm_query": str(stt_response.content)}
-        )
-        data = llm_response.json()
-        inner_data = json.loads(data["response_text"])
-
-        # Now, you can access the data from the inner JSON
-        risposta = inner_data.get("risposta")
-        stato = inner_data.get("stato")
-        tipologia = inner_data.get("tipologia")
-
-        print(risposta)
-        st.write("risposta: ", risposta)
-        st.write("stato: ", stato)
-        st.write("tipologia: ", tipologia)
-
-        # TTS
-        url = "http://localhost:8000/tts_query/"
-        out_path = requests.post(url=url, params={"input_text": str(risposta)})
-        print(out_path.json())
-        play_mp3(out_path.json())

src/main.py

@@ -17,6 +17,7 @@ with open("conf/train_llm.yaml", "r") as file_in:
     cfg = yaml.safe_load(file_in)
 
 # read system message
+# TODO: download from wandb
 with open("data/system_template.txt", "r") as file_in:
     system_message = file_in.read()
 

src/models/openai_llm.py

@@ -109,8 +109,8 @@ def run_query_on_wandb(cfg: DictConfig):
                 "token_usage": res["token_usage"],
                 "model_name": cfg.openai_parameters.model,
             },
-            start_time_ms=res["start_time_ms"],
-            end_time_ms=res["end_time_ms"],
+            # start_time_ms=res["start_time_ms"],
+            # end_time_ms=res["end_time_ms"],
             inputs={
                 "query": query,
                 "system_prompt": system_message,

src/pages/page1.py

@@ -0,0 +1,61 @@
+import streamlit as st
+from utils import Recorder, record_audio, play_mp3
+import os
+import requests
+import ast
+import json
+from openai import OpenAI
+import random
+from dotenv import load_dotenv
+import time
+
+load_dotenv()
+api_key = os.getenv("OPENAI_API_KEY")
+
+# st.title("PinocchioLand!")
+
+# # Display an image
+# st.image(
+#     "https://i.pinimg.com/736x/30/e9/36/30e936e18912e9a5670b88ec94630b4a.jpg",
+#     use_column_width=True,
+# )
+
+
+client = OpenAI(api_key=api_key)
+
+st.title("Chat with Pinocchio")
+
+if "count1" not in st.session_state:
+    st.session_state.count1 = 0
+
+if "openai_model" not in st.session_state:
+    st.session_state["openai_model"] = "gpt-3.5-turbo"
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+n = len(st.session_state.messages)
+
+if prompt := st.chat_input("What is up?"):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    with st.chat_message("assistant"):
+        # LLM
+        url = "http://localhost:8000/llm_query/"
+        # Append the query parameter to the URL
+        llm_response = requests.post(url=url, params={"llm_query": str(prompt)})
+        data = llm_response.json()
+        inner_data = json.loads(data["response_text"])
+
+        # Now, you can access the data from the inner JSON
+        risposta = inner_data.get("risposta")
+        stato = inner_data.get("stato")
+        tipologia = inner_data.get("tipologia")
+        st.write(inner_data)
+        st.session_state.messages.append({"role": "assistant", "content": inner_data})
+        # st.session_state.count1 += 1

src/pages/page2.py

@@ -0,0 +1,45 @@
+import streamlit as st
+import requests
+from utils import Recorder, record_audio, play_mp3
+import json
+
+st.title("Audio Pipeline")
+
+button_label = "record"
+
+if st.button(button_label):
+    st.write("recording...")
+    recording_path = record_audio()
+    button_label
+    with open(recording_path, "rb") as audio_file:
+        # Define the multipart/form-data payload
+        files = {"audio_file": (recording_path.split("/")[-1], audio_file, "audio/mp3")}
+
+        # Make the POST request
+        stt_response = requests.post("http://localhost:8000/stt_query/", files=files)
+        st.write("domanda :", stt_response.json())
+
+        # LLM
+        url = "http://localhost:8000/llm_query/"
+        # Append the query parameter to the URL
+        llm_response = requests.post(
+            url=url, params={"llm_query": str(stt_response.content)}
+        )
+        data = llm_response.json()
+        inner_data = json.loads(data["response_text"])
+
+        # Now, you can access the data from the inner JSON
+        risposta = inner_data.get("risposta")
+        stato = inner_data.get("stato")
+        tipologia = inner_data.get("tipologia")
+
+        print(risposta)
+        st.write("risposta: ", risposta)
+        st.write("stato: ", stato)
+        st.write("tipologia: ", tipologia)
+
+        # TTS
+        url = "http://localhost:8000/tts_query/"
+        out_path = requests.post(url=url, params={"input_text": str(risposta)})
+        print(out_path.json())
+        play_mp3(out_path.json())