Hugging Face's logo

Spaces:
BramLeo
/
adminzy
Paused

App Files Community
BramLeo commited on
Commit
717d9bc
·
verified ·
1 Parent(s): 249a2af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +65 -23
app.py CHANGED
@@ -31,7 +31,50 @@ def initialize_settings(model_path):
31
  # Fungsi untuk Menginisialisasi Index
32
  def initialize_index():
33
  # Tentukan dokumen input untuk pembacaan data
34
- documents = SimpleDirectoryReader(input_files=["bahandokumen/target.txt"]).load_data()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
35
 
36
  parser = SentenceSplitter(chunk_size=150, chunk_overlap=10)
37
  nodes = parser.get_nodes_from_documents(documents)
@@ -53,9 +96,6 @@ def initialize_chat_engine(index):
53
 
54
  # Fungsi untuk menghasilkan respons chatbot
55
  def generate_response(message, history, chat_engine):
56
- if history is None: # ✅ Cek jika history None
57
- history = []
58
-
59
  chat_messages = [
60
  ChatMessage(
61
  role="system",
@@ -66,35 +106,37 @@ def generate_response(message, history, chat_engine):
66
  "Chatbot: Syarat kerja di perusahaan meliputi waktu kerja, gaji, dan tunjangan sesuai dengan peraturan perusahaan."
67
  ),
68
  ]
69
-
70
  response = chat_engine.stream_chat(message)
71
  text = "".join(response.response_gen) # Gabungkan semua token menjadi string
72
-
73
- history.append((message, text)) # ✅ Sekarang history sudah pasti berupa list
74
  return history
75
 
76
  def clear_history(chat_engine):
77
  chat_engine.clear()
78
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
79
  def main():
80
  # Unduh model dan inisialisasi pengaturan
81
  model_path = initialize_llama_model()
82
- initialize_settings(model_path)
83
-
84
- # Inisialisasi index dan chat engine
85
  index = initialize_index()
86
- chat_engine = initialize_chat_engine(index)
87
-
88
- # Fungsi untuk chat
89
- def chatbot_response(message, history):
90
- return generate_response(message, history, chat_engine)
91
-
92
- # Luncurkan Gradio UI
93
- gr.Interface(
94
- fn=chatbot_response,
95
- inputs=["text"],
96
- outputs=["text"],
97
- ).launch()
98
-
99
  if __name__ == "__main__":
100
  main()
 
31
  # Fungsi untuk Menginisialisasi Index
32
  def initialize_index():
33
  # Tentukan dokumen input untuk pembacaan data
34
+ documents = SimpleDirectoryReader(input_files=["bahandokumen/K3.txt",
35
+ "bahandokumen/bonus.txt",
36
+ "bahandokumen/absensi.txt",
37
+ "bahandokumen/cuti.txt",
38
+ "bahandokumen/disiplinkerja.txt",
39
+ "bahandokumen/fasilitas&bantuan.txt",
40
+ "bahandokumen/fasilitaskerja.txt",
41
+ "bahandokumen/hak.txt",
42
+ "bahandokumen/hubunganpengusaha&serikat.txt",
43
+ "bahandokumen/istilah.txt",
44
+ "bahandokumen/jaminanserikat.txt",
45
+ "bahandokumen/jamkes.txt",
46
+ "bahandokumen/jamsos.txt",
47
+ "bahandokumen/keluhkesah.txt",
48
+ "bahandokumen/kenaikanupah.txt",
49
+ "bahandokumen/kewajiban.txt",
50
+ "bahandokumen/kompensasi.txt",
51
+ "bahandokumen/larangan.txt",
52
+ "bahandokumen/lembur.txt",
53
+ "bahandokumen/luaskesepakatan.txt",
54
+ "bahandokumen/mogok.txt",
55
+ "bahandokumen/pelanggaran&sanksi.txt",
56
+ "bahandokumen/pendidikan.txt",
57
+ "bahandokumen/pengangkatan.txt",
58
+ "bahandokumen/penilaian&promosi.txt",
59
+ "bahandokumen/pensiun.txt",
60
+ "bahandokumen/perjadin.txt",
61
+ "bahandokumen/pesangon.txt",
62
+ "bahandokumen/phk.txt",
63
+ "bahandokumen/pihak.txt",
64
+ "bahandokumen/pkb.txt",
65
+ "bahandokumen/resign.txt",
66
+ "bahandokumen/sanksi.txt",
67
+ "bahandokumen/shift.txt",
68
+ "bahandokumen/syaratkerja.txt",
69
+ "bahandokumen/sisacuti.txt",
70
+ "bahandokumen/target.txt",
71
+ "bahandokumen/tatacara.txt",
72
+ "bahandokumen/tka.txt",
73
+ "bahandokumen/tunjangan.txt",
74
+ "bahandokumen/uangpisah.txt",
75
+ "bahandokumen/upah.txt",
76
+ "bahandokumen/upahlembur.txt",
77
+ "bahandokumen/waktukerja.txt"]).load_data()
78
 
79
  parser = SentenceSplitter(chunk_size=150, chunk_overlap=10)
80
  nodes = parser.get_nodes_from_documents(documents)
 
96
 
97
  # Fungsi untuk menghasilkan respons chatbot
98
  def generate_response(message, history, chat_engine):
 
 
 
99
  chat_messages = [
100
  ChatMessage(
101
  role="system",
 
106
  "Chatbot: Syarat kerja di perusahaan meliputi waktu kerja, gaji, dan tunjangan sesuai dengan peraturan perusahaan."
107
  ),
108
  ]
 
109
  response = chat_engine.stream_chat(message)
110
  text = "".join(response.response_gen) # Gabungkan semua token menjadi string
111
+ history.append((message, text)) # Tambahkan ke riwayat
 
112
  return history
113
 
114
  def clear_history(chat_engine):
115
  chat_engine.clear()
116
 
117
+ # Inisialisasi Komponen Gradio untuk UI
118
+ def launch_gradio(chat_engine):
119
+ with gr.Blocks() as demo:
120
+ # Mengatur tombol untuk menghapus riwayat chat
121
+ clear_btn = gr.Button("Clear")
122
+ clear_btn.click(lambda: clear_history(chat_engine))
123
+
124
+ # Membuat antarmuka chat
125
+ chat_interface = gr.ChatInterface(
126
+ lambda message, history: generate_response(message, history, chat_engine)
127
+ )
128
+ demo.launch()
129
+
130
+ # Fungsi Utama untuk Menjalankan Aplikasi
131
  def main():
132
  # Unduh model dan inisialisasi pengaturan
133
  model_path = initialize_llama_model()
134
+ initialize_settings(model_path) # Mengirimkan model_path ke fungsi initialize_settings
135
+ # Inisialisasi index dan engine
 
136
  index = initialize_index()
137
+ chat_engine = initialize_chat_engine(index)
138
+ # Luncurkan antarmuka
139
+ launch_gradio(chat_engine)
140
+
 
 
 
 
 
 
 
 
 
141
  if __name__ == "__main__":
142
  main()