Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| from transcription import deepgram | |
| from summary import summarize | |
| from langchain.callbacks.base import BaseCallbackHandler | |
| from langchain_cohere import ChatCohere | |
| class StreamHandler(BaseCallbackHandler): | |
| def __init__(self, container, initial_text=""): | |
| self.container = container | |
| self.text = initial_text | |
| def on_llm_new_token(self, token: str, **kwargs) -> None: | |
| self.text += token | |
| self.container.markdown(self.text) | |
| def main(): | |
| session_state = st.session_state | |
| if 'transcription' not in session_state: | |
| session_state.transcription = "" | |
| if 'summary' not in session_state: | |
| session_state.summary = "" | |
| st.title("Meeting Insights") | |
| audio_file = st.sidebar.file_uploader("Upload your audio file:") | |
| language = st.sidebar.selectbox( | |
| "Select Language", | |
| ("English", "French", "Spanish")) | |
| if audio_file: | |
| if language: | |
| st.sidebar.audio(audio_file) | |
| if st.sidebar.button("Generate Transcription"): | |
| if language=="English": | |
| session_state.transcription=deepgram(audio_file , language="en") | |
| elif language=="French": | |
| session_state.transcription=deepgram(audio_file,language="fr") | |
| elif language=="Spanish": | |
| session_state.transcription=deepgram(audio_file , language="es") | |
| with st.container(height=500 , border=True): | |
| st.markdown(session_state.transcription) | |
| st.download_button('Download Transcription', session_state.transcription , file_name="Transcription.txt") | |
| if st.sidebar.toggle("Generate Summary"): | |
| st.header("Summary of the meeting") | |
| with st.container(height=500): | |
| stream_handler = StreamHandler(st.empty()) | |
| llm = ChatCohere(temperature = 0 ,streaming=True ,model = "command-r-plus" , callbacks=[stream_handler]) | |
| summarize(session_state.transcription , llm) | |
| # import clipboard | |
| # if st.sidebar.toggle("Generate Summary"): | |
| # st.markdown(""" | |
| # <style> | |
| # .big-font { | |
| # font-size:30px !important; | |
| # } | |
| # </style> | |
| # """, unsafe_allow_html=True) | |
| # st.markdown('<p class="big-font">Summary of the meeting</p>', unsafe_allow_html=True) | |
| # stream_handler = StreamHandler(st.empty()) | |
| # llm = ChatCohere(temperature = 0 ,streaming=True ,model = "command-r-plus" , callbacks=[stream_handler]) | |
| # data=summarize(session_state.transcription , llm) | |
| # summary_text = data.content | |
| # st.write(summary_text) | |
| # copy_button = st.button("Copy Summary to Clipboard") | |
| # if copy_button: | |
| # clipboard.copy(summary_text) | |
| # st.success("Summary copied to clipboard!") | |
| if __name__ == "__main__": | |
| main() | |