simplify UI

app.py

@@ -21,8 +21,7 @@ def chat_with_groq(message):
         messages=[
             {
                 "role": "system",
-                "content": """Anda adalah asisten medis yang membantu dokter dalam menyusun catatan SOAP berdasarkan percakapan dokter dan pasien.
-                            Ringkaskan dalam bentuk paragraf tanpa adanya bullet point dan gunakan bahasa Indonesia.""",
+                "content": """Anda adalah asisten medis yang membantu dokter dalam menyusun catatan medis dalam bentuk paragraf menggunakan bahasa Indonesia.""",
             },
             {"role": "user", "content": message},
         ],
@@ -39,7 +38,7 @@ def save_to_file(content, filename):
 
 
 def transcribe_audio(audio_file):
-    """Transkripsi audio menggunakan Faster Whisper tanpa koreksi model Hugging Face."""
+    """Transkripsi audio menggunakan Whisper tanpa koreksi model Hugging Face."""
     # segments, _ = model.transcribe(audio_file)
     # raw_transcription = " ".join(segment.text for segment in segments)
     with open(audio_file, "rb") as file:
@@ -50,33 +49,36 @@ def transcribe_audio(audio_file):
         )
         print(res)
         raw_transcription = res.text
+        soap_output, download_soap = generate_soap_summary(raw_transcription)
+        tags_output, download_tags = detect_medical_tags(raw_transcription)
     return (
-        raw_transcription,
-        save_to_file(raw_transcription, "transcription_large.txt"),
+        save_to_file(raw_transcription, "raw_transcription.txt"),
         audio_file,
+        soap_output,
+        download_soap,
+        tags_output,
+        download_tags,
     )
 
 
-def generate_soap_summary(transcription_text, selected_model):
-    """Membuat ringkasan SOAP dari teks transkripsi menggunakan model yang dipilih."""
-    template = """Harap buat ringkasan dalam format berikut:
+def generate_soap_summary(transcription_text):
+    template = """Buat ringkasan SOAP berdasarkan percakapan dokter dan pasien dalam format berikut:
 
     Subjective:
     ICD10:
     Objective:
     Assessment:
     Plan:
+
     ### Percakapan:
     {dialogue}
-
-    ### Catatan SOAP:
     """
 
     soap = chat_with_groq(template.format(dialogue=transcription_text))
     return soap, save_to_file(soap, "soap_summary.txt")
 
 
-def detect_medical_tags(transcription_text, selected_model):
+def detect_medical_tags(transcription_text):
     """Mendeteksi tags Diagnosis, Obat, Hasil Lab, dan Radiologi menggunakan model yang dipilih."""
     template = """
     Identifikasi dan berikan saran dalam bahasa Indonesia tindakan logis selanjutnya dalam format:
@@ -94,24 +96,16 @@ def detect_medical_tags(transcription_text, selected_model):
 
 # Antarmuka Gradio
 with gr.Blocks(
-    title="AI-based Medical SOAP Summarization and Tag Detection with Faster Whisper Large"
+    title="AI-based Medical SOAP Summarization and Tag Detection with Whisper Large"
 ) as app:
-    gr.Markdown(
-        "## Medical SOAP Summarization and Tag Detection with Faster Whisper Large"
-    )
+    gr.Markdown("## Medical SOAP Summarization and Tag Detection with Whisper Large")
 
     with gr.Row():
         with gr.Column():
-            audio_input = gr.Audio("microphone", type="filepath", label="🎙️ Rekam Suara")
-            transcribe_button = gr.Button("🎧 Transkripsi dengan Whisper Large")
-            soap_button = gr.Button("📝 Buat SOAP")
-            tags_button = gr.Button("🏷️ Deteksi Saran Tags")
-            transcription_edit_box = gr.Textbox(
-                label="📄 Hasil Transkripsi (Faster Whisper Large) - Bisa Diedit",
-                lines=3,
-                interactive=True,
+            audio_input = gr.Audio(
+                source="microphone", type="filepath", label="🎙️ Rekam Suara"
             )
-            update_transcription_button = gr.Button("💾 Simpan Hasil Edit")
+            transcribe_button = gr.Button("🎧 Tulis Rekam Medis")
 
         with gr.Column():
             soap_output = gr.Textbox(label="📃 Hasil SOAP", lines=10, interactive=False)
@@ -129,28 +123,14 @@ with gr.Blocks(
     transcribe_button.click(
         transcribe_audio,
         inputs=[audio_input],
-        outputs=[transcription_edit_box, download_transcription, download_audio],
-    )
-
-    # Tombol Simpan Hasil Edit
-    update_transcription_button.click(
-        lambda text: (text, save_to_file(text, "user_edited_transcription.txt")),
-        inputs=[transcription_edit_box],
-        outputs=[transcription_edit_box, download_transcription],
-    )
-
-    # Tombol SOAP
-    soap_button.click(
-        generate_soap_summary,
-        inputs=[transcription_edit_box],
-        outputs=[soap_output, download_soap],
-    )
-
-    # Tombol Tags
-    tags_button.click(
-        detect_medical_tags,
-        inputs=[transcription_edit_box],
-        outputs=[tags_output, download_tags],
+        outputs=[
+            download_transcription,
+            download_audio,
+            soap_output,
+            download_soap,
+            tags_output,
+            download_tags,
+        ],
     )
 
 # Jalankan aplikasi