Update app.py

app.py

@@ -13,10 +13,11 @@ import os
 
 print(os.listdir('.'))
 
+
 # Download the 'punkt' package
 nltk.download('punkt')
 
-openai.api_key = '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'
+#openai.api_key = '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'
 
 # Define your model architecture
 class Bert_model(nn.Module):
@@ -95,6 +96,33 @@ def explain_term(term):
 
 # Streamlit code to upload file
 st.title('FinQA (Financial Question-Answering)')
+
+api_key = st.text_input("Enter your OpenAI API key:", type="password")
+
+if api_key:
+    try:
+        openai.api_key = api_key
+
+        # Test the API key by making a small request
+        openai.ChatCompletion.create(
+            model="gpt-4",
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": "Hello"},
+            ],
+        )
+
+        # If the above code doesn't raise an exception, the API key is valid
+        st.success("API key is valid!")
+
+        # Continue with the rest of your code...
+        uploaded_file = st.file_uploader("Choose a PDF file", type="pdf")
+        
+    except Exception as e:
+        st.error(f"Failed to validate API key: {e}")
+else:
+    st.warning("Please enter your OpenAI API key.")
+    
 uploaded_file = st.file_uploader("Choose a PDF file", type="pdf")
 
 # Select language
@@ -136,7 +164,7 @@ if uploaded_file is not None:
         chat_history = "\n".join([sentence[0] for sentence in top_5_sentences])
 
         # Ask the question using OpenAI API
-        openai.api_key = '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'  # Replace with your actual OpenAI API key
+        #openai.api_key = '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'  # Replace with your actual OpenAI API key
 
         response = openai.ChatCompletion.create(
             model="gpt-4",