Upload bot.py

bot.py

@@ -4,10 +4,9 @@ import os
 import re
 import time
 import base64
-from patterns import count_patterns, list_patterns, pdf_request_patterns, greetings,farewell # Import patterns
+from patterns import count_patterns, list_patterns, pdf_request_patterns, greetings, farewell  # Import patterns
 from google.cloud import storage
-
-os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "/Users/ninadmandavkar/Downloads/fynd_prod.json"
+from tempfile import NamedTemporaryFile
 
 # Base GCS path and local download path
 BASE_GCS_PATH = "gs://fynd-assets-private/documents/daytrader/"
@@ -64,7 +63,6 @@ def typing_effect(text):
     typed_text_placeholder.markdown(styled_text, unsafe_allow_html=True)
 
 def chatbot_response(user_input):
-
     lower_input = user_input.lower()
 
     if any(greet in lower_input for greet in greetings):
@@ -126,66 +124,95 @@ def download_pdf(file_path):
     return base64.b64encode(pdf_data).decode('utf-8')  # Encode to Base64 and decode to string
 
 def main():
-    st.markdown(
-    """
-    <h1 style="
-        background-image: linear-gradient(to right, #800000, #ff0000); 
-        -webkit-background-clip: text; 
-        -webkit-text-fill-color: transparent; 
-        font-size: 3rem;  /* Adjust size as needed */
-        text-align: center; /* Center the title */
-    ">
-        Fynder
-    </h1>
-    """, 
+    # st.markdown(
+    # """
+    # <h1 style="
+    #     background-image: linear-gradient(to right, #800000, #ff0000); 
+    #     -webkit-background-clip: text; 
+    #     -webkit-text-fill-color: transparent; 
+    #     font-size: 3rem;  /* Adjust size as needed */
+    #     text-align: center; /* Center the title */
+    # ">
+    #     Fynder
+    # </h1>
+    # """, 
+    # unsafe_allow_html=True
+    # )
+
+    # Upload JSON credentials
+    uploaded_file = st.file_uploader("Upload your fynd_prod.json file", type="json")
+    
+    if uploaded_file is not None:
+        # Save the file temporarily
+        with NamedTemporaryFile(delete=False, suffix=".json") as temp_file:
+            temp_file.write(uploaded_file.read())
+            json_path = temp_file.name
+        
+        os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = json_path
+
+        st.markdown(
+            """
+            <h1 style="
+                background-image: linear-gradient(to right, #800000, #ff0000); 
+                -webkit-background-clip: text; 
+                -webkit-text-fill-color: transparent; 
+                font-size: 3rem;  /* Adjust size as needed */
+                text-align: center; /* Center the title */
+            ">
+                Fynder
+            </h1>
+            """, 
     unsafe_allow_html=True
-    )
-
-    user_input = st.text_input("", value="", help="Type your prompt here")
-
-    if user_input:
-        response = chatbot_response(user_input)
-
-        if isinstance(response, tuple):
-            bot_response, pdf_name = response
-            typing_effect(bot_response)
-
-            result = fetch_pdf_from_gcs(pdf_name)
-            if result.returncode == 0:
-                downloaded_files = [f for f in os.listdir(LOCAL_DOWNLOAD_PATH) if pdf_name in f and f.endswith('.pdf')]
-                
-                if downloaded_files:
-                    for pdf_file in downloaded_files:
-                        file_path = os.path.join(LOCAL_DOWNLOAD_PATH, pdf_file)  # Update this path as per your structure
-                        pdf_data = download_pdf(file_path)
-
-                        # Create a custom download button with gradient styling
-                        download_button_html = f"""
-                        <a href="data:application/pdf;base64,{pdf_data}" download="{pdf_file}" 
-                        style="
-                                display: inline-block; 
-                                padding: 10px 20px; 
-                                color: white; 
-                                text-decoration: none; 
-                                border-radius: 20px; 
-                                background-image: linear-gradient(to right, #800000, #ff0000); 
-                                box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-                            ">
-                            Download {pdf_file}
-                        </a>
-                        """
-                        st.markdown(download_button_html, unsafe_allow_html=True)
+        )
+        
+        
+        user_input = st.text_input("", value="", help="Type your prompt here")
+
+        if user_input:
+            response = chatbot_response(user_input)
+
+            if isinstance(response, tuple):
+                bot_response, pdf_name = response
+                typing_effect(bot_response)
+
+                result = fetch_pdf_from_gcs(pdf_name)
+                if result.returncode == 0:
+                    downloaded_files = [f for f in os.listdir(LOCAL_DOWNLOAD_PATH) if pdf_name in f and f.endswith('.pdf')]
+                    
+                    if downloaded_files:
+                        for pdf_file in downloaded_files:
+                            file_path = os.path.join(LOCAL_DOWNLOAD_PATH, pdf_file)
+                            pdf_data = download_pdf(file_path)
+
+                            # Create a custom download button with gradient styling
+                            download_button_html = f"""
+                            <a href="data:application/pdf;base64,{pdf_data}" download="{pdf_file}" 
+                            style="
+                                    display: inline-block; 
+                                    padding: 10px 20px; 
+                                    color: white; 
+                                    text-decoration: none; 
+                                    border-radius: 20px; 
+                                    background-image: linear-gradient(to right, #800000, #ff0000); 
+                                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+                                ">
+                                Download {pdf_file}
+                            </a>
+                            """
+                            st.markdown(download_button_html, unsafe_allow_html=True)
+                    else:
+                        typing_effect("No files matching that name were found.")
                 else:
-                    typing_effect("No files matching that name were found.")
-            else:
-                typing_effect("There was an error fetching the PDF. Please check the name and try again.")
-                st.write(result.stderr)
+                    typing_effect("There was an error fetching the PDF. Please check the name and try again.")
+                    st.write(result.stderr)
 
-        elif isinstance(response, list):  # When the response is a list of files
-            for pdf_file in response:
-                typing_effect(pdf_file)  # Apply typing effect to each file
-        else:
-            typing_effect(response)
+            elif isinstance(response, list):  # When the response is a list of files
+                for pdf_file in response:
+                    typing_effect(pdf_file)  # Apply typing effect to each file
+            else:
+                typing_effect(response)
+    else:
+        st.markdown("")
 
 if __name__ == "__main__":
     main()