app.py

app.py

@@ -17,7 +17,7 @@ st.caption("Chat with your contract or manage meeting minutes")
 
 # Sidebar for API Key input
 with st.sidebar:
-    OPENAI_API_KEY = st.text_input("Enter your C2 Group of Technologies provided key", type="password")
+    OPENAI_API_KEY = st.text_input("Enter your OpenAI API Key", type="password")
 
 # Tabs for Contract and Minutes
 tab1, tab2 = st.tabs(["Contract", "Minutes"])
@@ -31,7 +31,7 @@ with tab1:
     if OPENAI_API_KEY:
         client = OpenAI(api_key=OPENAI_API_KEY)
     else:
-        st.error("Please enter your C2 Group of Technologies provided key to continue.")
+        st.error("Please enter your OpenAI API key to continue.")
         st.stop()
 
     ASSISTANT_ID = "asst_rd9h8PfYuOmHbkvOF3RTmVfn"
@@ -111,20 +111,28 @@ with tab2:
             st.text_area("Meeting Transcript", combined_text, height=300)
             
             if st.button("Generate Meeting Minutes"):
-                response = client.chat.completions.create(
-                    model="gpt-4-turbo",
-                    messages=[
-                        {"role": "system", "content": "You are an AI assistant that generates professional meeting minutes."},
-                        {"role": "user", "content": f"Summarize the following into structured meeting minutes:\n{combined_text}"}
-                    ]
+                response = client.beta.threads.messages.create(
+                    thread_id=ASSISTANT_ID,
+                    role="user",
+                    content=f"Summarize the following into structured meeting minutes:\n{combined_text}"
                 )
-                minutes = response.choices[0].message.content
+                
+                run = client.beta.threads.runs.create(
+                    thread_id=ASSISTANT_ID,
+                    assistant_id=ASSISTANT_ID
+                )
+                
+                while True:
+                    run_status = client.beta.threads.runs.retrieve(thread_id=ASSISTANT_ID, run_id=run.id)
+                    if run_status.status == "completed":
+                        break
+                    time.sleep(1)
+                
+                messages = client.beta.threads.messages.list(thread_id=ASSISTANT_ID)
+                minutes = messages.data[0].content[0].text.value
                 st.write("### Meeting Minutes:")
                 st.text_area("Generated Minutes", minutes, height=300)
 
-                date_stamp = datetime.now().strftime("%Y-%m-%d")
-                file_name = f"Minutes_{date_stamp}.docx"
-                
                 doc = Document()
                 doc.add_paragraph(minutes)
                 
@@ -134,5 +142,30 @@ with tab2:
                 
                 st.download_button(label="Download Meeting Minutes", 
                                    data=docx_io, 
-                                   file_name=file_name, 
-                                   mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
+                                   file_name="Meeting_Minutes.docx", 
+                                   mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
+                
+                if st.button("Contract Analysis"):
+                    analysis_prompt = "Analyse this data and provide any insights, risks and relevant data based on the contract that we need to be aware of.\n" + minutes
+                    response = client.beta.threads.messages.create(
+                        thread_id=ASSISTANT_ID,
+                        role="user",
+                        content=analysis_prompt
+                    )
+                    
+                    run = client.beta.threads.runs.create(
+                        thread_id=ASSISTANT_ID,
+                        assistant_id=ASSISTANT_ID
+                    )
+                    
+                    while True:
+                        run_status = client.beta.threads.runs.retrieve(thread_id=ASSISTANT_ID, run_id=run.id)
+                        if run_status.status == "completed":
+                            break
+                        time.sleep(1)
+                    
+                    messages = client.beta.threads.messages.list(thread_id=ASSISTANT_ID)
+                    analysis_output = messages.data[0].content[0].text.value
+                    
+                    st.write("### Contract Analysis Output:")
+                    st.text_area("Contract Analysis", analysis_output, height=300)