Create app.py

app.py

@@ -0,0 +1,63 @@
+import streamlit as st
+from pyannote.audio import Pipeline
+import tempfile
+import os
+
+# Set page title
+st.set_page_config(page_title="Speaker Diarization App")
+
+st.title("Speaker Diarization App")
+
+# File uploader
+uploaded_file = st.file_uploader("Choose an audio file", type=['wav', 'mp3'])
+
+# Hugging Face access token input
+hf_token = st.text_input("Enter your Hugging Face access token:", type="password")
+
+if uploaded_file is not None and hf_token:
+    # Save uploaded file temporarily
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.wav') as tmp_file:
+        tmp_file.write(uploaded_file.getvalue())
+        tmp_path = tmp_file.name
+
+    # Instantiate the pipeline
+    @st.cache_resource
+    def load_pipeline(token):
+        return Pipeline.from_pretrained(
+            "pyannote/speaker-diarization-3.1",
+            use_auth_token=token
+        )
+
+    try:
+        pipeline = load_pipeline(hf_token)
+
+        # Run the pipeline on the audio file
+        with st.spinner('Processing audio...'):
+            diarization = pipeline(tmp_path)
+
+        # Generate RTTM content
+        rttm_content = ""
+        for turn, _, speaker in diarization.itertracks(yield_label=True):
+            rttm_line = f"SPEAKER {os.path.basename(tmp_path)} 1 {turn.start:.3f} {turn.duration:.3f} <NA> <NA> {speaker} <NA> <NA>\n"
+            rttm_content += rttm_line
+
+        # Display RTTM content
+        st.subheader("Diarization Results (RTTM format)")
+        st.text_area("RTTM Output", rttm_content, height=300)
+
+        # Provide download button for RTTM file
+        st.download_button(
+            label="Download RTTM file",
+            data=rttm_content,
+            file_name="diarization.rttm",
+            mime="text/plain"
+        )
+
+    except Exception as e:
+        st.error(f"An error occurred: {str(e)}")
+
+    # Clean up the temporary file
+    os.unlink(tmp_path)
+
+else:
+    st.info("Please upload an audio file and enter your Hugging Face access token to start.")