app.py

app.py

@@ -5,77 +5,89 @@ import time
 st.set_page_config(page_title="Schlager ContractAi")
 
 st.title("Schlager ContractAi")
-st.caption("Chat with your contract")
+st.caption("Chat with your contract or manage meeting minutes")
 
-# Sidebar for API Key input and file uploads
+# Sidebar for API Key input
 with st.sidebar:
     OPENAI_API_KEY = st.text_input("Enter your OpenAI API Key", type="password")
-    st.subheader("Minutes Uploads")
-    uploaded_files = st.file_uploader("Upload meeting minutes (PDF/DOCX)", type=["pdf", "docx"], accept_multiple_files=True)
 
-    if uploaded_files:
-        st.write("### Uploaded Files:")
-        for uploaded_file in uploaded_files:
-            st.write(f"- {uploaded_file.name}")
+# Tabs for Contract and Minutes
+tab1, tab2 = st.tabs(["Contract", "Minutes"])
 
-# Initialize OpenAI client
-if OPENAI_API_KEY:
-    client = OpenAI(api_key=OPENAI_API_KEY)
-else:
-    st.error("Please enter your OpenAI API key to continue.")
-    st.stop()
+with tab1:
+    st.subheader("Contract Chat")
+    
+    # Initialize OpenAI client
+    if OPENAI_API_KEY:
+        client = OpenAI(api_key=OPENAI_API_KEY)
+    else:
+        st.error("Please enter your OpenAI API key to continue.")
+        st.stop()
 
-ASSISTANT_ID = "asst_rd9h8PfYuOmHbkvOF3RTmVfn"
+    ASSISTANT_ID = "asst_rd9h8PfYuOmHbkvOF3RTmVfn"
 
-# Session state for chat messages
-if "messages" not in st.session_state:
-    st.session_state["messages"] = []
+    # Session state for chat messages
+    if "messages" not in st.session_state:
+        st.session_state["messages"] = []
 
-# Display all chat messages
-for message in st.session_state.messages:
-    role, content = message["role"], message["content"]
-    if role == "assistant" and "|" in content:  # Detect table formatting
-        st.chat_message(role).write(content, unsafe_allow_html=True)
-    else:
-        st.chat_message(role).write(content)
-
-# User input
-if prompt := st.chat_input():
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    st.chat_message("user").write(prompt)
-
-    # Send message to OpenAI Assistant
-    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
-
-        # Detect table format and use Markdown
-        if "|" in assistant_message and "---" in assistant_message:
-            formatted_message = f"```\n{assistant_message}\n```"
-            st.chat_message("assistant").write(formatted_message, unsafe_allow_html=True)
+    # Display all chat messages
+    for message in st.session_state.messages:
+        role, content = message["role"], message["content"]
+        if role == "assistant" and "|" in content:  # Detect table formatting
+            st.chat_message(role).write(content, unsafe_allow_html=True)
         else:
-            st.chat_message("assistant").write(assistant_message)
+            st.chat_message(role).write(content)
+
+    # User input
+    if prompt := st.chat_input():
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        st.chat_message("user").write(prompt)
+
+        # Send message to OpenAI Assistant
+        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
+            )
 
-        st.session_state.messages.append({"role": "assistant", "content": assistant_message})
+            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)
 
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+            messages = client.beta.threads.messages.list(thread_id=thread_id)
+            assistant_message = messages.data[0].content[0].text.value
+
+            # Detect table format and use Markdown
+            if "|" in assistant_message and "---" in assistant_message:
+                formatted_message = f"```
+{assistant_message}
+```"
+                st.chat_message("assistant").write(formatted_message, unsafe_allow_html=True)
+            else:
+                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:
+    st.subheader("Minutes")
+    st.write("Hello World")
+    
+    uploaded_files = st.file_uploader("Upload meeting minutes (PDF/DOCX)", type=["pdf", "docx"], accept_multiple_files=True)
+    
+    if uploaded_files:
+        st.write("### Uploaded Files:")
+        for uploaded_file in uploaded_files:
+            st.write(f"- {uploaded_file.name}")