Update app.py

app.py

@@ -5,12 +5,13 @@ from sentence_transformers import SentenceTransformer
 import faiss
 from groq import Groq
 import json
+import re
 
 # Set up Groq client
 client = Groq(api_key=os.environ.get('GroqApi'))
 
-# Load the model
-model = SentenceTransformer('all-MiniLM-L6-v2')  # Use an open-source embedding model
+# Load the embedding model
+model = SentenceTransformer('all-MiniLM-L6-v2')  # Open-source embedding model
 
 # Initialize FAISS
 dimension = 384  # Embedding dimension for 'all-MiniLM-L6-v2'
@@ -80,11 +81,27 @@ if uploaded_file:
                 messages=[{"role": "user", "content": json.dumps(query_content)}],
                 model="llama3-8b-8192",
             )
-            tariff_rate = float(chat_completion.choices[0].message.content.strip())
 
-            # Calculate and display bill
-            total_bill = tariff_rate * total_units
-            st.write(f"Tariff Rate: {tariff_rate} PKR/kWh")
-            st.write(f"Total Bill: {total_bill:.2f} PKR")
-
-# Run Streamlit using: streamlit run app.py
+            # Extract numeric value from the response
+            response_content = chat_completion.choices[0].message.content.strip()
+
+            try:
+                # Attempt to parse the response as JSON or extract the numeric value
+                if response_content.startswith('{') and response_content.endswith('}'):
+                    # Parse JSON if valid
+                    response_data = json.loads(response_content)
+                    tariff_rate = response_data.get('tariff_rate', 0.0)  # Replace 'tariff_rate' with correct key
+                else:
+                    # Extract numeric value using regex
+                    match = re.search(r"[-+]?\d*\.\d+|\d+", response_content)
+                    tariff_rate = float(match.group()) if match else 0.0
+
+                # Calculate and display the bill
+                total_bill = tariff_rate * total_units
+                st.write(f"Tariff Rate: {tariff_rate} PKR/kWh")
+                st.write(f"Total Bill: {total_bill:.2f} PKR")
+
+            except ValueError as e:
+                st.error(f"Error parsing Groq response: {e}")
+                st.write("Groq response content:")
+                st.write(response_content)