update testing script and rollback chat

app/chat.py

@@ -21,34 +21,18 @@ def initialize_session_state():
         st.session_state['should_speak'] = True
     if 'input_text' not in st.session_state:
         st.session_state['input_text'] = ""
-    if 'audio_cache' not in st.session_state:
-        st.session_state['audio_cache'] = {}
-    if 'play_audio' not in st.session_state:
-        st.session_state['play_audio'] = None
+    if 'tts_output' not in st.session_state:
+        st.session_state['tts_output'] = ""
 
-# Fungsi TTS dengan cache
-def text_to_speech(text, message_id=None):
-    # Cek cache terlebih dahulu
-    if message_id and message_id in st.session_state['audio_cache']:
-        return st.session_state['audio_cache'][message_id]
-    
+# Fungsi TTS
+def text_to_speech(text):
     tts = gtts.gTTS(text, lang="id")
     audio_bytes = BytesIO()
     tts.write_to_fp(audio_bytes)
     audio_bytes.seek(0)
     audio_base64 = base64.b64encode(audio_bytes.read()).decode()
-    
-    # Simpan ke cache jika ada message_id
-    if message_id:
-        st.session_state['audio_cache'][message_id] = audio_base64
-    
-    return audio_base64
-
-# Fungsi untuk membuat audio player
-def create_audio_player(audio_base64, autoplay=False):
-    autoplay_attr = "autoplay" if autoplay else ""
     audio_player = f"""
-    <audio {autoplay_attr} controls style="width: 100%; margin-top: 5px;">
+    <audio autoplay>
         <source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3">
     </audio>
     """
@@ -108,42 +92,23 @@ def display_chat_history(chain):
         st.session_state['generated'].append(output)
         st.session_state.input_text = ""
 
-        # Auto-play TTS untuk pesan baru jika diaktifkan
+        # Simpan output untuk TTS
         if st.session_state['should_speak'] and output:
-            message_id = len(st.session_state['generated']) - 1
-            audio_base64 = text_to_speech(output, message_id)
-            st.session_state['play_audio'] = message_id
+            st.session_state['tts_output'] = output
+            if 'tts_played' in st.session_state:
+                del st.session_state['tts_played']
 
-    # Tampilkan riwayat chat dengan tombol TTS
+    # Tampilkan riwayat chat
     if st.session_state['generated']:
         with reply_container:
             for i in range(len(st.session_state['generated'])):
-                # Tampilkan pesan user
                 message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="no-avatar")
-                
-                # Tampilkan pesan bot
                 message(st.session_state["generated"][i], key=str(i), avatar_style="no-avatar")
-                
-                # Tombol dan audio player untuk pesan bot
-                col1, col2 = st.columns([1, 4])
-                with col1:
-                    if st.button(f"🔊", key=f"play_tts_{i}", help="Putar ulang audio"):
-                        st.session_state['play_audio'] = i
-                        st.experimental_rerun()
-                
-                # Tampilkan audio player jika pesan ini yang dipilih untuk diputar
-                if st.session_state.get('play_audio') == i:
-                    with col2:
-                        message_text = st.session_state["generated"][i]
-                        audio_base64 = text_to_speech(message_text, i)
-                        
-                        # Auto-play untuk pesan baru, manual untuk replay
-                        autoplay = (i == len(st.session_state['generated']) - 1 and 
-                                  st.session_state['should_speak'])
-                        
-                        audio_player = create_audio_player(audio_base64, autoplay)
-                        st.markdown(audio_player, unsafe_allow_html=True)
-                    
-                    # Reset play_audio setelah ditampilkan
-                    if st.session_state.get('play_audio') == i:
-                        st.session_state['play_audio'] = None
+
+    # Tampilkan dan mainkan audio TTS jika ada
+    if st.session_state.get('tts_output') and not st.session_state.get('tts_played'):
+        st.markdown(text_to_speech(st.session_state['tts_output']), unsafe_allow_html=True)
+        st.session_state['tts_played'] = True
+    elif st.session_state.get('tts_played'):
+        st.session_state['tts_output'] = ""
+        del st.session_state['tts_played']

evaluate.py

@@ -61,7 +61,7 @@ class PNPRAGEvaluator:
         # Initialize RAG chain
         self.chain = ConversationalRetrievalChain.from_llm(
             self.llm, 
-            retriever=self.vector_store.as_retriever(search_kwargs={"k": 6}),
+            retriever=self.vector_store.as_retriever(search_kwargs={"k": 10}),
             combine_docs_chain_kwargs={"prompt": sahabat_prompt},
             return_source_documents=True, 
             memory=self.memory
@@ -496,7 +496,7 @@ evaluation_dataset = [
         "ground_truth": "Politeknik Negeri Padang menjalin kerjasama dengan industri, pemerintah, BUMN, dan asosiasi profesi baik dalam negeri maupun luar negeri. Bentuk kerjasama mencakup rekrutmen, prakerin (praktik kerja industri), kunjungan industri, bimbingan karir, serta pembuatan MoU. Tujuannya untuk menjaga mutu lulusan dan penyaluran SDM."
     },
     {
-        "question": "Siapa saja pimpinan di Politeknik Negeri Padang saat ini?",
+        "question": "Siapa saja direktur di Politeknik Negeri Padang saat ini?",
         "ground_truth": "Direktur Politeknik Negeri Padang adalah Dr. Ir. Surfa Yondri, S.T., S.ST., M.Kom. Wakil Direktur Bidang Akademik adalah Ir. Revalin Herdianto, ST., M.Sc., Ph.D. Pimpinan lainnya antara lain Nasrullah, ST., M.T., dan Sarmiadi, S.E., M.M. yang memiliki pengalaman panjang dalam jabatan struktural di kampus."
     },
     {

evaluate_ragas.py

@@ -0,0 +1,132 @@
+from datasets import Dataset
+import json
+from datetime import datetime
+from ragas.metrics import (
+    answer_relevancy,
+    context_precision,
+    context_recall,
+    ContextRelevance,
+    faithfulness
+)
+from ragas import evaluate
+from app.document_processor import load_vector_store_from_supabase
+from app.db import supabase
+from app.prompts import sahabat_prompt
+from langchain.chains import ConversationalRetrievalChain
+from langchain.memory import ConversationBufferMemory
+from langchain_community.llms import Replicate
+import os
+
+# Load vector store
+BUCKET_NAME = "pnp-bot-storage-archive"
+VECTOR_STORE_PREFIX = "vector_store"
+vector_store = load_vector_store_from_supabase(supabase, BUCKET_NAME, VECTOR_STORE_PREFIX)
+
+# RAG Chain
+def create_rag_chain():
+    llm = Replicate(
+        model="fauziisyrinapridal/sahabat-ai-v1:afb9fa89fe786362f619fd4fef34bd1f7a4a4da23073d8a6fbf54dcbe458f216",
+        model_kwargs={"temperature": 0.1, "top_p": 0.9, "max_new_tokens": 10000},
+        replicate_api_token=os.getenv("REPLICATE_API_TOKEN"),
+    )
+    memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True, output_key='answer')
+    chain = ConversationalRetrievalChain.from_llm(
+        llm,
+        retriever=vector_store.as_retriever(search_kwargs={"k": 10}),
+        combine_docs_chain_kwargs={"prompt": sahabat_prompt},
+        return_source_documents=True,
+        memory=memory
+    )
+    return chain
+
+chain = create_rag_chain()
+
+# PNP Evaluation Dataset
+evaluation_dataset = [
+    {
+        "question": "Bagaimana sistem pendidikan yang diterapkan di Politeknik Negeri Padang?",
+        "ground_truth": "Sistem pendidikan yang diterapkan di Politeknik adalah dengan menggabungkan pendidikan teoritis, praktek (terapan) di Laboratorium dan praktek industry. Pelaksanaan praktik di industri dilakukan oleh mahasiswa selama satu semester untuk menambah wawasan, pengalaman dan pengembangan ilmu guna membentuk tenaga ahli yang terampil dan profesional."
+    },
+    {
+        "question": "Apa saja mata kuliah yang terdapat dalam kurikulum pendidikan Politeknik Negeri Padang?",
+        "ground_truth": "Kurikulum Pendidikan telah disusun berbasis kompetensi dengan kelompok mata kuliah sebagai berikut : - Mata Kuliah Pengembangan Kepribadian (MPK) - Mata Kuliah Keimuan dan Keterampilan (MKK) - Mata Kuliah Berkarya (MKB) - Mata Kuliah Berkehidupan Bermasyarakat (MBB)"
+    },
+    {
+        "question": "Bagaimana Politeknik Negeri Padang mendukung misi tridharma perguruan tinggi?",
+        "ground_truth": "Politeknik Negeri Padang dalam menjalankan misi tridharma perguruan tinggi didukung oleh tenaga pendidik dan tenaga kependidikan yang profesional pada bidangnya. Jumlah dan kualifikasi staf tersebut berdasarkan keadaan Desember 2017 sebagai berikut : - Tenaga Pendidik : S1 = 14 orang, S2 = 256 orang, S3 = 21 orang (Yang sedang menempuh S3 = 7 orang, Yang sedang menempuh S2 = 5 orang) - Tenaga Kependidikan : SD = 5 orang, SMP = 4 orang, SLTA = 71 orang, D3 = 25 orang, S1 = 54 orang, S2 = 15 orang."
+    },
+    {
+        "question": "Bagaimana Politeknik Negeri Padang menyediakan akses internet bagi mahasiswa?",
+        "ground_truth": "Politeknik Negeri Padang telah memiliki Anjungan Internet Mandiri (AIM) yang dapat diakses oleh mahasiswa secara gratis, yang tersedia pada titik-titik strategis. Juga tersedia kawasan hot spot area di sekitar kampus sehingga mahasiswa dapat memanfaatkan internet dengan bebas menggunakan laptop/PC."
+    },
+    {
+        "question": "Apa saja contoh kerjasama Politeknik Negeri Padang dengan industri?",
+        "ground_truth": "PT. Siemens Indonesia, PT. Toyota Aichi Takaoua Japan, PT. PLN, PT. INTI, Futaba Rashi Siisha Kusho Japan, PT. Sintom, PT. Krakatau Steel, Komatssu Shinge Koumuten, PT. PAL Indonesia, PT. Hexindo, Taishurin Co. Ltd Fukuoaka Japan, PT. Texmaco Perkasa, PT. LEN Industri, PT. Toyota Astra Motor, PT. Indah Kiat, PT. Trakindo Utama, BTN."
+    },
+    {
+        "question": "Bagaimana Politeknik Negeri Padang membantu mahasiswa dalam bidang prestasi dan ekonomi?",
+        "ground_truth": "Tersedia bantuan untuk sekitar 800 mahasiswa setiap tahunnya. Beasiswa yang diterima antara lain: - Beasiswa Peningkatan Prestasi Akademik (PPA), - Beasiswa Kerja Mahasiswa (BKM), - Beasiswa Bantuan Belajar Mahasiswa (BBM), - Beasiswa TPSDP, - Beasiswa Kredit Bantuan Belajar Mahasiswa (KBBM), - Beasiswa Depertemen Hankam (ABRI), - Beasiswa PT. Toyota Astra, - Beasiswa ORBIT (ICMI), - Beasiswa Supersemar."
+    },
+    {
+        "question": "Bagaimana status akreditasi program studi di Politeknik Negeri Padang?",
+        "ground_truth": "Program studi di Politeknik Negeri Padang memiliki status akreditasi yang bervariasi seperti Baik, Baik Sekali, hingga Unggul. Contohnya, Teknik Mesin (D3) terakreditasi Baik Sekali hingga 2029, Teknik Manufaktur (D4) terakreditasi Unggul hingga 2028, dan Teknik Sipil (D3) terakreditasi A hingga 2026. Setiap program memiliki SK dan sertifikat akreditasi resmi."
+    },
+    {
+        "question": "Bagaimana proses penerimaan mahasiswa baru di Politeknik Negeri Padang?",
+        "ground_truth": "Penerimaan mahasiswa baru di Politeknik Negeri Padang dilakukan melalui berbagai jalur seleksi seperti SNBT, SNMPN, dan kelas kerjasama. Tersedia brosur dan informasi detail melalui situs http://penerimaan.pnp.ac.id. Program studi Teknik Alat Berat misalnya memiliki kelas kerjasama dengan PT Trakindo Utama. Jadwal seleksi dan pengumuman dapat diakses secara daring."
+    },
+    {
+        "question": "Apa bentuk kerjasama yang dilakukan Politeknik Negeri Padang?",
+        "ground_truth": "Politeknik Negeri Padang menjalin kerjasama dengan industri, pemerintah, BUMN, dan asosiasi profesi baik dalam negeri maupun luar negeri. Bentuk kerjasama mencakup rekrutmen, prakerin (praktik kerja industri), kunjungan industri, bimbingan karir, serta pembuatan MoU. Tujuannya untuk menjaga mutu lulusan dan penyaluran SDM."
+    },
+    {
+        "question": "Siapa saja direktur di Politeknik Negeri Padang saat ini?",
+        "ground_truth": "Direktur Politeknik Negeri Padang adalah Dr. Ir. Surfa Yondri, S.T., S.ST., M.Kom. Wakil Direktur Bidang Akademik adalah Ir. Revalin Herdianto, ST., M.Sc., Ph.D. Pimpinan lainnya antara lain Nasrullah, ST., M.T., dan Sarmiadi, S.E., M.M. yang memiliki pengalaman panjang dalam jabatan struktural di kampus."
+    },
+    {
+        "question": "Bagaimana sejarah singkat berdirinya Politeknik Negeri Padang?",
+        "ground_truth": "Politeknik Negeri Padang didirikan pada tahun 1987 sebagai salah satu dari 17 politeknik pertama di Indonesia. Awalnya bernama Politeknik Engineering Universitas Andalas. Pada 1997 menjadi Politeknik Universitas Andalas lalu berubah menjadi Politeknik Negeri Padang. Saat ini memiliki 32 program studi dari jenjang D3 hingga Magister Terapan."
+    }
+]
+
+# Format Ragas
+ragas_data = {
+    "question": [],
+    "answer": [],
+    "contexts": [],
+    "ground_truth": []
+}
+
+for item in evaluation_dataset:
+    result = chain({"question": item["question"]})
+    documents = result.get("source_documents", [])
+    doc_texts = [doc.page_content for doc in documents]
+
+    ragas_data["question"].append(item["question"])
+    ragas_data["answer"].append(result["answer"])
+    ragas_data["contexts"].append(doc_texts)
+    ragas_data["ground_truth"].append(item["ground_truth"])
+
+# Buat dataset HuggingFace
+dataset = Dataset.from_dict(ragas_data)
+
+# Jalankan evaluasi
+result = evaluate(
+    dataset=dataset,
+    metrics=[
+        answer_relevancy,
+        context_precision,
+        context_recall,
+        ContextRelevance,
+        faithfulness
+    ]
+)
+
+print(result)
+# Simpan hasil evaluasi ke file JSON
+output_file = f"ragas_eval_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
+
+with open(output_file, "w", encoding="utf-8") as f:
+    json.dump(result, f, indent=2, ensure_ascii=False)
+
+print(f"Hasil evaluasi disimpan ke {output_file}")