costs

bill.py

@@ -78,12 +78,14 @@ def parseBill(data):
 
 
     return {
-        "billDate": billDate,
-        "billNo": billNo,
-        "amountDue": amountDue,
-        "extraCharge": extraCharge,
-        "totalBillCosts": totalBillCosts,
-        "subscriberCosts": subscriberCosts,
+        "Data Factura": billDate,
+        "Serie numar factura": billNo,
+        "De plata": amountDue,
+        "Costuri suplimentare": extraCharge,
+        "Total plata factura": totalBillCosts,
+        "Costuri utilizatori": subscriberCosts,
+        "Lista categorii": list(categories),
+        "Lista costuri": list(names)
 
     }
 
@@ -129,6 +131,29 @@ def process_query(query, user_id, model_name):
 
     return context
 
+# import the datetime class from the datetime module
+from datetime import datetime
+def log_conversation(user_id, user_query, assistant_response, tokens, cost):
+    log_entry = {
+        "timestamp": datetime.now().isoformat(),
+        "user_id": user_id,
+        "user_query": user_query,
+        "assistant_response": assistant_response,
+        "tokens": tokens,
+        "cost": cost
+    }
+    log_file_path = os.path.join("logs", "conversation_logs.json")
+    os.makedirs(os.path.dirname(log_file_path), exist_ok=True)
+    if os.path.exists(log_file_path):
+        with open(log_file_path, "r") as log_file:
+            logs = json.load(log_file)
+    else:
+        logs = []
+    logs.append(log_entry)
+    with open(log_file_path, "w") as log_file:
+        json.dump(logs, log_file, indent=4)
+
+
 def main():
     st.title("Telecom Bill Chat with LLM Agent")
 
@@ -140,41 +165,54 @@ def main():
     user_id = st.sidebar.text_input("Introdu numărul de telefon:")
     # display the user data if the user_id is set
     #st.write(f"User ID: {user_id}")
-
+    
+    st.session_state.user_data = None
+   
     if user_id and user_id != st.session_state.user_id:
         data = load_user_data(user_id)
+        st.session_state.user_data = data
         if data:
             st.session_state.user_id = user_id
             st.success("Utilizator găsit!")
+            st.write(f"Numar telefon: {st.session_state.user_id}")
+            st.session_state.user_data = data
+
         else:
-            st.warning("Nu am găsit date pentru acest ID. Încărcați o factură json.")
+            st.warning("Nu am găsit date pentru acest ID.")
+            st.warning("Încărcați o factură json.")
             st.session_state.user_id = user_id
+            st.session_state.user_data = None
+
+    # If the user has no data yet Show the upload (st.file_uploader...) dialog , 
+    # If the user has stored data in data\user_data\"user_data{user_id}.json, display the existing bills data - st.write(bill) but compacted  
+    if st.session_state.user_data:
+        st.write("Facturi existente (extras):")
+        for bill in st.session_state.user_data.get("bills", []):
+            st.write({
+                "Data factura": bill.get("billDate"),
+                "Numar serie factura": bill.get("billNo"),
+                "De plata": bill.get("amountDue"),
+                "Costuri suplimentare": bill.get("extraCharge")
+            })
+    if not st.session_state.user_data:
+        uploaded_file = st.file_uploader("Incarca factura", type="json")
+        if uploaded_file and st.session_state.user_id:
+            bill_data = json.load(uploaded_file)
+            parsed_bill = parseBill(bill_data)
+            existing_data = load_user_data(st.session_state.user_id)
+            
+            # Check if the billNo already exists in the existing data
+            existing_bill_nos = [bill.get("billNo") for bill in existing_data.get("bills", [])]
+            if parsed_bill.get("billNo") in existing_bill_nos:
+                st.warning("Factură existentă.")
+            else:
+                if "bills" not in existing_data:
+                    existing_data["bills"] = []
+                existing_data["bills"].append(parsed_bill)
+                save_user_data(st.session_state.user_id, existing_data)
+                st.success("Factura a fost încărcată și salvată cu succes!")
+     
 
-    uploaded_file = st.file_uploader("Upload JSON Bill", type="json")
-    if uploaded_file and st.session_state.user_id:
-        bill_data = json.load(uploaded_file)
-        parsed_bill = parseBill(bill_data)
-        existing_data = load_user_data(st.session_state.user_id)
-        
-        # Check if the billNo already exists in the existing data
-        existing_bill_nos = [bill.get("billNo") for bill in existing_data.get("bills", [])]
-        if parsed_bill.get("billNo") in existing_bill_nos:
-            st.warning("Factura existentă.")
-        else:
-            if "bills" not in existing_data:
-                existing_data["bills"] = []
-            existing_data["bills"].append(parsed_bill)
-            save_user_data(st.session_state.user_id, existing_data)
-            st.success("Factura a fost încărcată și salvată cu succes!")
-
-    if st.session_state.user_id:
-        data = load_user_data(st.session_state.user_id)
-        st.write(f"Numar telefon: {st.session_state.user_id}")
-        st.write("Facturi existente:")
-        for bill in data.get("bills", []):
-            st.write(bill)
-    else:
-        st.info("Introduceți un ID și/sau încărcați o factură JSON pentru a continua.")
 
     # Initialize conversation in the session state
     # "context_prompt_added" indicates whether we've added the specialized "bill info" context yet.
@@ -193,7 +231,7 @@ def main():
 
     if prompt := st.chat_input("Introduceți întrebarea aici:"):
         if not st.session_state.user_id:
-            st.error("Trebuie să introduceți un număr de telefon valid sau să încărcați date.")
+            st.error("Trebuie să introduci un număr de telefon valid sau să încarci date.")
             return
 
         # If the context prompt hasn't been added yet, build & inject it once;
@@ -225,5 +263,32 @@ def main():
             st.write("Completion tokens:", completion.usage.completion_tokens)
             st.write("Total tokens:", completion.usage.total_tokens)
 
+            # Estimate cost per conversation (find the OpenAI costs for gpt-4o and gpt-4o-mini model per token)
+            prompt_tokens = completion.usage.prompt_tokens
+            completion_tokens = completion.usage.completion_tokens
+            total_tokens = completion.usage.total_tokens
+
+            # Estimate cost per conversation
+            if model_name == "gpt-4o":
+                cost_per_token = 0.03 / 1000  # $0.03 per 1,000 tokens
+            elif model_name == "gpt-4o-mini":
+                cost_per_token = 0.015 / 1000  # $0.015 per 1,000 tokens
+
+            estimated_cost = total_tokens * cost_per_token
+            st.write("Estimated cost:", estimated_cost)
+
+            # Log the conversation
+            log_conversation(
+                st.session_state["user_id"],
+                prompt,
+                response_text,
+                {
+                    "prompt_tokens": prompt_tokens,
+                    "completion_tokens": completion_tokens,
+                    "total_tokens": total_tokens
+                },
+                estimated_cost
+            )
+
 if __name__ == "__main__":
     main()