Update app.py

app.py

@@ -4,29 +4,51 @@ import os
 import time
 
 # Load your OpenAI API key from the environment variable
-openai.api_key = os.getenv("OPENAI_API_KEY")
+api_key = os.getenv("OPENAI_API_KEY")
+
+# Ensure the API key doesn't have any leading/trailing whitespace or newline characters
+openai.api_key = api_key.strip()
+
+# Memory to store the conversation history
+conversation_history = []
 
 # Function to interact with the GPT-4o-mini model via the OpenAI API
 def subsurface_sentinel(user_input):
-    prompt = f"User: {user_input}"
-    
+    # Add user input to the conversation history
+    conversation_history.append({"role": "user", "content": user_input})
+
+    # Prepare the messages for the API call
+    messages = [
+        {"role": "system", "content": "You are Subsurface Sentinel, a virtual assistant for professionals in the oil and gas industry, focusing on field development planning."}
+    ] + conversation_history
+
     # Retry mechanism in case of connection errors
     for _ in range(3):  # Try up to 3 times
         try:
             response = openai.chat.completions.create(
                 model="gpt-4o-mini",  # Use the gpt-4o-mini model
-                messages=[
-                    {"role": "system", "content": "You are Subsurface Sentinel, a virtual assistant for professionals in the oil and gas industry, focusing on field development planning."},
-                    {"role": "user", "content": prompt}
-                ],
+                messages=messages,
                 max_tokens=150,
                 temperature=0.7,
             )
             # Extract the response text and remove any leading/trailing whitespace
-            return response.choices[0].message.content.strip()
+            assistant_message = response.choices[0].message['content'].strip()
+            
+            # Add assistant response to the conversation history
+            conversation_history.append({"role": "assistant", "content": assistant_message})
+            
+            return assistant_message
         except openai.error.APIConnectionError as e:
             print(f"Connection error: {e}. Retrying...")
             time.sleep(2)  # Wait for 2 seconds before retrying
+        except openai.error.InvalidRequestError as e:
+            return f"Invalid request: {e}"
+        except openai.error.AuthenticationError as e:
+            return f"Authentication failed: {e}"
+        except openai.error.RateLimitError as e:
+            return f"Rate limit exceeded: {e}"
+        except Exception as e:
+            return f"An unexpected error occurred: {e}"
     
     return "Failed to connect to OpenAI API after several attempts."