app.py

app.py

@@ -19,66 +19,75 @@ st.caption("Chat with your contract or manage meeting minutes")
 with st.sidebar:
     OPENAI_API_KEY = st.text_input("Enter your C2 Group of Technologies Access Key", type="password")
 
-# Tabs for Contract and Minutes
-tab1, tab2 = st.tabs(["Contract", "Minutes"])
+# Tabs for Contract, Technical, and Minutes
+tab1, tab2, tab3 = st.tabs(["Contract", "Technical", "Minutes"])
 
 SUPPORTED_AUDIO_FORMATS = (".mp3", ".wav", ".m4a")
 SUPPORTED_TEXT_FORMATS = (".txt", ".docx", ".csv", ".xlsx", ".pdf")
 
-with tab1:
-    st.subheader("Contract Chat")
-    
-    if OPENAI_API_KEY:
-        client = OpenAI(api_key=OPENAI_API_KEY)
-    else:
-        st.error("Please enter your C2 Group of Technologies Access Key to continue.")
-        st.stop()
 
-    ASSISTANT_ID = "asst_rd9h8PfYuOmHbkvOF3RTmVfn"
-    
-    if "messages" not in st.session_state:
-        st.session_state["messages"] = []
-    
-    if st.button("Clear Chat"):
-        st.session_state["messages"] = []
-        st.rerun()
-    
-    for message in st.session_state.messages:
-        role, content = message["role"], message["content"]
-        st.chat_message(role).write(content)
-
-    if prompt := st.chat_input():
-        st.session_state.messages.append({"role": "user", "content": prompt})
-        st.chat_message("user").write(prompt)
-
-        try:
-            thread = client.beta.threads.create()
-            thread_id = thread.id
-            client.beta.threads.messages.create(
-                thread_id=thread_id,
-                role="user",
-                content=prompt
-            )
-            
-            run = client.beta.threads.runs.create(
-                thread_id=thread_id,
-                assistant_id=ASSISTANT_ID
-            )
-
-            while True:
-                run_status = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
-                if run_status.status == "completed":
-                    break
-                time.sleep(1)
-
-            messages = client.beta.threads.messages.list(thread_id=thread_id)
-            assistant_message = messages.data[0].content[0].text.value
-            st.chat_message("assistant").write(assistant_message)
-            st.session_state.messages.append({"role": "assistant", "content": assistant_message})
-        except Exception as e:
-            st.error(f"Error: {str(e)}")
-
-with tab2:
+def contract_chat_section(tab, assistant_id):
+    with tab:
+        st.subheader("Chat")
+        
+        if OPENAI_API_KEY:
+            client = OpenAI(api_key=OPENAI_API_KEY)
+        else:
+            st.error("Please enter your C2 Group of Technologies Access Key to continue.")
+            st.stop()
+
+        if "messages" not in st.session_state:
+            st.session_state["messages"] = []
+
+        if st.button("Clear Chat"):
+            st.session_state["messages"] = []
+            st.rerun()
+
+        for message in st.session_state.messages:
+            role, content = message["role"], message["content"]
+            st.chat_message(role).write(content)
+
+        if prompt := st.chat_input():
+            st.session_state.messages.append({"role": "user", "content": prompt})
+            st.chat_message("user").write(prompt)
+
+            try:
+                thread = client.beta.threads.create()
+                thread_id = thread.id
+                client.beta.threads.messages.create(
+                    thread_id=thread_id,
+                    role="user",
+                    content=prompt
+                )
+
+                run = client.beta.threads.runs.create(
+                    thread_id=thread_id,
+                    assistant_id=assistant_id
+                )
+
+                while True:
+                    run_status = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
+                    if run_status.status == "completed":
+                        break
+                    time.sleep(1)
+
+                messages = client.beta.threads.messages.list(thread_id=thread_id)
+                assistant_message = messages.data[0].content[0].text.value
+                st.chat_message("assistant").write(assistant_message)
+                st.session_state.messages.append({"role": "assistant", "content": assistant_message})
+            except Exception as e:
+                st.error(f"Error: {str(e)}")
+
+ASSISTANT_CONTRACT_ID = "asst_rd9h8PfYuOmHbkvOF3RTmVfn"
+ASSISTANT_TECHNICAL_ID = "asst_technical_example"  # Replace with actual assistant ID
+
+# Contract Chat Section
+contract_chat_section(tab1, ASSISTANT_CONTRACT_ID)
+
+# Technical Chat Section
+contract_chat_section(tab2, ASSISTANT_TECHNICAL_ID)
+
+with tab3:
     st.subheader("Minutes")
     
     if "generated_minutes" not in st.session_state:
@@ -128,23 +137,19 @@ with tab2:
                 **Date:** [Enter date]  
                 **Time:** [Enter time]  
                 **Attendees:** [List attendees]  
-
                 ### Agenda Items
                 - [Agenda Item 1]
                 - [Agenda Item 2]
                 - [Agenda Item 3]
-
                 ### Action Items
                 | Action Item | Owner(s) | Deadline | Status |
                 |------------|---------|----------|--------|
                 | [Action Item 1] | [Owner(s) 1] | [Deadline 1] | [Status 1] |
                 | [Action Item 2] | [Owner(s) 2] | [Deadline 2] | [Status 2] |
                 | [Action Item 3] | [Owner(s) 3] | [Deadline 3] | [Status 3] |
-
                 ---
                 **Meeting Summary:**  
                 [Brief summary of key points, discussions, and decisions]
-
                 **Transcript:**  
                 {combined_text}
                 """
@@ -160,43 +165,3 @@ with tab2:
                 
                 st.write("### Generated Meeting Minutes:")
                 st.text_area("Meeting Minutes", st.session_state["generated_minutes"], height=400)
-            
-            if st.session_state["generated_minutes"]:
-                doc = Document()
-                doc.add_paragraph(st.session_state["generated_minutes"])
-                
-                docx_io = BytesIO()
-                doc.save(docx_io)
-                docx_io.seek(0)
-                
-                st.download_button(label="Download Meeting Minutes", 
-                                   data=docx_io, 
-                                   file_name="Meeting_Minutes.docx", 
-                                   mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
-
-                if st.button("Contract Analysis"):
-                    analysis_prompt = "Analyze this data and provide insights, risks, and relevant data based on the contract:\n" + st.session_state["generated_minutes"]
-                    thread = client.beta.threads.create()
-                    thread_id = thread.id
-                    client.beta.threads.messages.create(
-                        thread_id=thread_id,
-                        role="user",
-                        content=analysis_prompt
-                    )
-
-                    run = client.beta.threads.runs.create(
-                        thread_id=thread_id,
-                        assistant_id=ASSISTANT_ID
-                    )
-
-                    while True:
-                        run_status = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
-                        if run_status.status == "completed":
-                            break
-                        time.sleep(1)
-
-                    messages = client.beta.threads.messages.list(thread_id=thread_id)
-                    analysis_output = messages.data[0].content[0].text.value
-
-                    st.write("### Contract Analysis Output:")
-                    st.text_area("Contract Analysis", analysis_output, height=300)