changes summary added

README.md

@@ -1,13 +0,0 @@
----
-title: Erginous Project
-emoji: 🚀
-colorFrom: blue
-colorTo: green
-sdk: streamlit
-sdk_version: 1.34.0
-app_file: app.py
-pinned: false
-license: apache-2.0
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,6 +1,19 @@
 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():
@@ -8,8 +21,8 @@ def main():
     
     if 'transcription' not in session_state:
         session_state.transcription = ""
-    if "summary" not in session_state:
-        session_state.summary=""
+    if 'summary' not in session_state:
+        session_state.summary = ""
 
 
 
@@ -34,15 +47,59 @@ def main():
 
     
     
+    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!")
+
+
+
+
 
-    transcription=st.text_area("Transcription",value=session_state.transcription, height=400)
-    if transcription:
-        st.download_button('Download Transcription', session_state.transcription , file_name="Transcription.txt") 
-        if st.sidebar.toggle("Generate Summary"):
-            session_state.summary=summarize(session_state.transcription)
-            st.text_area("Summary" , value=session_state.summary , height=400)
 
 if __name__ == "__main__":
     main()
 
-

summary.py

@@ -1,19 +1,20 @@
-from langchain_groq import ChatGroq
+from langchain.callbacks.base import BaseCallbackHandler
+from langchain_cohere import ChatCohere
 from langchain.prompts import ChatPromptTemplate
-from langchain_core.output_parsers import StrOutputParser
-from dotenv import load_dotenv
+import streamlit as st
+from  dotenv import load_dotenv
 
 load_dotenv()
 
-model = ChatGroq(temperature=0, model_name="mixtral-8x7b-32768") #mixtral-8x7b-32768
 
 
-def summarize(transcription):
-    template="""You are provided  with transcription of a meeting between different people. Your task is to summarize this transcription
-    Transcription: {transcription}"""
 
+def summarize(transcription , llm):
+    template="""Please provide a brief and concise summary of the key discussion points and outcomes from the meeting transcription. Ensure that the summary captures the main topics, decisions made, action items, and any important next steps or deadlines agreed upon. 
+                The summary should be structured clearly and be easily understandable to those who were not present at the meeting.
+        Transcription: {transcription}"""
     prompt = ChatPromptTemplate.from_template(template)
-
-    chain = prompt | model | StrOutputParser()
-    summary=chain.invoke({"transcription":transcription})
-    return summary
+    chain = prompt | llm
+    response = chain.invoke({'transcription':transcription})
+    return response
+