Update app.py

app.py

@@ -2,6 +2,8 @@ import os
 import json
 import time
 import requests
+# Lägg till Anthropic import
+from anthropic import Anthropic
 from openai import OpenAI
 import gradio as gr
 from sentence_transformers import SentenceTransformer
@@ -23,11 +25,18 @@ RETRIEVAL_K = 3
 # Kontrollera om vi kör i Hugging Face-miljön
 IS_HUGGINGFACE = os.environ.get("SPACE_ID") is not None
 
+# OpenAI-klient behålls för bakåtkompatibilitet
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 if not OPENAI_API_KEY:
     raise ValueError("OPENAI_API_KEY saknas")
 client = OpenAI(api_key=OPENAI_API_KEY)
 
+# Lägg till Anthropic API-nyckel och klient
+ANTHROPIC_API_KEY = os.environ.get("ANTHROPIC_API_KEY")
+if not ANTHROPIC_API_KEY:
+    raise ValueError("ANTHROPIC_API_KEY saknas")
+anthropic_client = Anthropic(api_key=ANTHROPIC_API_KEY)
+
 log_folder = "logs"
 os.makedirs(log_folder, exist_ok=True)
 log_file_path = os.path.join(log_folder, "conversation_log_v2.txt")
@@ -198,27 +207,33 @@ def generate_answer(query):
     context, sources = retrieve_context(query)
     if not context.strip():
         return "Jag hittar ingen relevant information i mina källor.\n\nDetta är ett AI genererat svar."
+    
+    # System-prompts och användarfråga
     prompt = f"""{prompt_template}
     
 Relevant kontext:
 {context}
 Fråga: {query}
 Svar (baserat enbart på den indexerade datan):"""
+    
     try:
-        response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
+        # Använd Claude Haiku istället för GPT
+        response = anthropic_client.messages.create(
+            model="claude-3-5-haiku-20240307",
+            max_tokens=500,
+            temperature=0.2,
+            system=("Du är en expert på ChargeNodes produkter och tjänster. "
+                   "Svara enbart baserat på den information som finns i den indexerade datan."),
             messages=[
-                {"role": "system", "content": "Du är en expert på ChargeNodes produkter och tjänster. Svara enbart baserat på den information som finns i den indexerade datan."},
                 {"role": "user", "content": prompt}
-            ],
-            temperature=0.2,
-            max_tokens=500
+            ]
         )
-        answer = response.choices[0].message.content
+        answer = response.content[0].text
         return answer + "\n\nAI-genererat. Otillräcklig hjälp? Kontakta support@chargenode.eu eller 010-2051055"
     except Exception as e:
         return f"Tekniskt fel: {str(e)}\n\nAI-genererat. Kontakta support@chargenode.eu eller 010-2051055"
 
+# Resten av koden lämnas oförändrad...
 # --- Slack Integration ---
 def send_to_slack(subject, content, color="#2a9d8f"):
     """Basfunktion för att skicka meddelanden till Slack."""