Initial commit for hoax detector

app.py

@@ -13,15 +13,15 @@ def detect_hoax(text):
     vec = vectorizer.transform([text])
     result = model.predict(vec)[0]
     if result == 1:
-        return gr.update(value="HOAX", visible=True), gr.update(visible=True, elem_classes="hoax-red")
+        return "<div style='background-color:#e74c3c; color:white; padding:10px; border-radius:5px'>HOAX</div>"
     else:
-        return gr.update(value="Bukan Hoax", visible=True), gr.update(visible=True, elem_classes="hoax-green")
+        return "<div style='background-color:#27ae60; color:white; padding:10px; border-radius:5px'>BUKAN HOAX</div>"
 
-def qa_chat(message, history, context):
+def qa_manual(message, history, context):
     if not context:
-        return "Mohon masukkan teks berita di kolom atas terlebih dahulu."
+        return history + [[message, "Mohon isi teks berita terlebih dahulu."]]
     result = qa_pipe(question=message, context=context)
-    return result["answer"]
+    return history + [[message, result["answer"]]]
 
 def ner(text):
     entities = ner_pipe(text)
@@ -31,30 +31,33 @@ def ner(text):
     }
     for ent in entities:
         color = color_map.get(ent["entity_group"], "#eee")
-        styled += f"<mark style='background-color:{color}; padding: 2px; margin:2px'>{ent['word']} <small>({ent['entity_group']})</small></mark> "
+        styled += f"<mark style='background-color:{color}; padding:2px; margin:2px'>{ent['word']} <small>({ent['entity_group']})</small></mark> "
     return styled
 
 # --- UI Gradio ---
-with gr.Blocks(css=".hoax-red {color: white; background-color: #e74c3c; padding: 10px; border-radius: 5px;} .hoax-green {color: white; background-color: #27ae60; padding: 10px; border-radius: 5px;}") as demo:
+with gr.Blocks() as demo:
     gr.Markdown("## Hoax Detector App")
 
-    # Shared input
     context_input = gr.Textbox(label="Teks Berita / Konteks", lines=5, placeholder="Masukkan teks berita di sini...")
 
     with gr.Tab("Deteksi Hoaks"):
-        hoax_btn = gr.Button("DETEKSI")
-        hoax_result = gr.Textbox(visible=False, label="Hasil Deteksi", interactive=False, show_label=False)
-        hoax_result_style = gr.HTML(visible=False)
-        hoax_btn.click(fn=detect_hoax, inputs=context_input, outputs=[hoax_result, hoax_result_style])
+        detect_btn = gr.Button("DETEKSI")
+        hoax_output = gr.HTML()
+        detect_btn.click(fn=detect_hoax, inputs=context_input, outputs=hoax_output)
 
     with gr.Tab("QA"):
         #gr.Markdown("### Tanya Jawab Berdasarkan Teks Berita")
-        qa_chatbot = gr.ChatInterface(
-            fn=lambda msg, hist: qa_chat(msg, hist, context_input.value),
-            title="",
-            textbox=gr.Textbox(placeholder="Tanyakan sesuatu...", label=None),
-            submit_btn="KIRIM"
-        )
+        qa_question = gr.Textbox(placeholder="Tulis pertanyaan...", label="Pertanyaan")
+        qa_btn = gr.Button("KIRIM")
+        qa_history = gr.Chatbot(label="Riwayat Tanya Jawab")
+        qa_state = gr.State([])
+
+        qa_btn.click(
+            fn=qa_manual,
+            inputs=[qa_question, qa_state, context_input],
+            outputs=[qa_history],
+            show_progress=False
+        ).then(fn=lambda h: h, inputs=qa_history, outputs=qa_state)
 
     with gr.Tab("NER"):
         ner_btn = gr.Button("Ekstrak Entitas")