check secrets access

.gitignore

@@ -0,0 +1 @@
+.venv

app.py

@@ -1,4 +1,116 @@
 import streamlit as st
+import openai
+import anthropic
+import fitz
 
-x = st.slider("Select a value")
-st.write(x, "squared is", x * x)
+# Set up API clients
+openai_api_key = st.secrets["OPENAI_API_KEY"]
+openai_client = openai.OpenAI(api_key=openai_api_key)
+
+anthropic_api_key = st.secrets["ANTHROPIC_API_KEY"]
+anthropic_client = anthropic.Anthropic(api_key=anthropic_api_key)
+
+
+# Password protection
+def check_password():
+    if "password_correct" not in st.session_state:
+        st.session_state["password_correct"] = False
+
+    if not st.session_state["password_correct"]:
+        password = st.text_input("Enter the password", type="password")
+        if st.button("Submit"):
+            if password == st.secrets["PASSWORD"]:
+                st.session_state["password_correct"] = True
+                st.rerun()
+            else:
+                st.error("😕 Password incorrect")
+        return False
+    else:
+        return True
+
+
+def extract_text_from_pdf(file):
+    try:
+        with fitz.open(stream=file.read(), filetype="pdf") as doc:
+            text = ""
+            for page in doc:
+                text += page.get_text()
+        return text
+    except Exception as e:
+        st.error(f"Error parsing PDF: {str(e)}")
+        return None
+
+
+def main():
+    st.title("Harvey Legal Research Take-Home")
+
+    if not check_password():
+        return
+
+    # Model selector
+    model = st.selectbox("Select Model", ["GPT-4o", "Claude 3.5 Sonnet"])
+
+    # Prompt input
+    prompt = st.text_area("Enter your prompt:", height=200)
+
+    # Document upload
+    uploaded_file = st.file_uploader("Upload a document", type=["txt", "pdf"])
+    document_text = None
+    if uploaded_file is not None:
+        if uploaded_file.type == "text/plain":
+            document_text = uploaded_file.getvalue().decode()
+        elif uploaded_file.type == "application/pdf":
+            document_text = extract_text_from_pdf(uploaded_file)
+        else:
+            st.error("Unsupported file type. Please upload a text or PDF file.")
+
+    # Run button
+    if st.button("Run"):
+        if not prompt:
+            st.warning("Please enter a prompt.")
+            return
+
+        if not document_text:
+            st.warning("Please upload a document.")
+            return
+
+        # full_prompt = f"Document content:\n\n{document_text}\n\nPrompt:\n\n{prompt}"
+        full_prompt = prompt
+
+        if model == "GPT-4o":
+            with st.spinner("Processing with GPT-4o..."):
+                stream = openai_client.chat.completions.create(
+                    model="gpt-4o",
+                    messages=[{"role": "user", "content": full_prompt}],
+                    stream=True,
+                )
+
+                full_response = ""
+                response_area = st.empty()
+
+                for chunk in stream:
+                    if chunk.choices[0].delta.content is not None:
+                        full_response += chunk.choices[0].delta.content
+                        response_area.text_area(
+                            "Response:", value=full_response, height=300
+                        )
+
+        elif model == "Claude 3.5 Sonnet":
+            with st.spinner("Processing with Claude 3.5 Sonnet..."):
+                with anthropic_client.messages.stream(
+                    model="claude-3-5-sonnet-20240620",
+                    max_tokens=8192,
+                    messages=[{"role": "user", "content": full_prompt}],
+                ) as stream:
+                    full_response = ""
+                    response_area = st.empty()
+
+                    for text in stream.text_stream:
+                        full_response += text
+                        response_area.text_area(
+                            "Response:", value=full_response, height=300
+                        )
+
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,28 @@
+annotated-types==0.7.0
+anthropic==0.34.2
+anyio==4.4.0
+certifi==2024.8.30
+charset-normalizer==3.3.2
+distro==1.9.0
+exceptiongroup==1.2.2
+filelock==3.16.1
+fsspec==2024.9.0
+h11==0.14.0
+httpcore==1.0.5
+httpx==0.27.2
+huggingface-hub==0.25.0
+idna==3.10
+jiter==0.5.0
+openai==1.46.0
+packaging==24.1
+pydantic==2.9.2
+pydantic_core==2.23.4
+PyMuPDF==1.24.10
+PyMuPDFb==1.24.10
+PyYAML==6.0.2
+requests==2.32.3
+sniffio==1.3.1
+tokenizers==0.20.0
+tqdm==4.66.5
+typing_extensions==4.12.2
+urllib3==2.2.3