Update app.py

app.py

@@ -5,10 +5,7 @@ import json
 import os
 
 # Get the API key from the environment variable
-API_KEY = os.getenv('PERPLEXITY_API_KEY')
-
-# Define the API endpoint (same as before)
-API_URL = "https://api.perplexity.ai/v1/estimate-costs"
+API_KEY = os.getenv('PPX_KEY')
 
 # Function to get dollar estimates for damaged objects
 def get_damaged_object_estimates(damage_description):
@@ -17,31 +14,33 @@ def get_damaged_object_estimates(damage_description):
         st.error("API key not found. Make sure it's set in the environment variables.")
         return None
 
-    # Prepare the request payload
-    payload = {
-        "description": damage_description  # Sending the full string as a description
-    }
+    # Messages to send 
+    messages = [
+        {
+            "role": "system",
+            "content": (
+                "You are an artificial intelligence assistant to help hurricane-afflicted homeowners file for insurance."
+            ),
+        },
+        {
+            "role": "user",
+            "content": (
+                f"Can you use this general list of damages to my home to prepare an organized, itemized list with dollar estimates that I can submit to my insurance company? \n\n {damage_description}"
+            ),
+        },
+    ]
 
-    # Prepare headers with the API key
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {API_KEY}"
-    }
+    # instantiate client to make Perplexity requests
+    client = OpenAI(api_key=API_KEY, base_url="https://api.perplexity.ai")
 
-    # Make the API request
-    response = requests.post(API_URL, headers=headers, data=json.dumps(payload))
+    # chat completion without streaming
+    response = client.chat.completions.create(
+        model="llama-3-sonar-large-32k-online",
+        messages=messages,
+    )
+
+    return response
 
-    # Check for a successful response
-    if response.status_code == 200:
-        # Parse the JSON response
-        response_data = response.json()
-        
-        # Return the itemized list with estimated costs
-        return response_data["itemized_list"]
-    else:
-        # Handle errors
-        st.error(f"Error: {response.status_code}, {response.text}")
-        return None
 
 # Streamlit interface
 st.title("Disaster Damage Cost Estimator")
@@ -58,9 +57,7 @@ if st.button("Get Estimated Costs"):
             
             if estimated_costs:
                 st.success("Here are the estimated costs:")
-                # Display the itemized list with dollar estimates
-                for item in estimated_costs:
-                    st.write(f"- {item['object']}: ${item['estimated_cost']}")
+                st.write(estimated_costs)
             else:
                 st.error("Failed to retrieve estimates.")
     else: