Update app.py

app.py

@@ -23,6 +23,9 @@ MAX_CHUNK_SIZE = 2000  # Ökad chunkstorleken för att bättre hantera FAQ-svar
 CHUNK_OVERLAP = 200    # Nytt: Overlapping chunks för att inte tappa kontext
 RETRIEVAL_K = 5        # Antal chunker att hämta vid varje sökning
 
+# Uppdaterad modell till Sonnet 4
+MODEL_NAME = "claude-sonnet-4-20250514"
+
 # Kontrollera om vi kör i Hugging Face-miljön
 IS_HUGGINGFACE = os.environ.get("SPACE_ID") is not None
 
@@ -363,18 +366,42 @@ def retrieve_context(query, k=RETRIEVAL_K):
 # Ladda prompt template
 prompt_template = load_prompt()
 
-def generate_answer(query):
-    """Genererar svar baserat på fråga och retrieval-baserad kontext med Claude Haiku."""
+def format_chat_history_for_claude(chat_history):
+    """Formaterar chatthistoriken för Claude API med endast de senaste meddelandena för att undvika token-gränser."""
+    # Ta endast de senaste 10 meddelandena för att hålla kontexten hanterbar
+    recent_history = chat_history[-10:] if len(chat_history) > 10 else chat_history
+    
+    messages = []
+    for msg in recent_history:
+        if msg["role"] in ["user", "assistant"]:
+            messages.append({
+                "role": msg["role"],
+                "content": msg["content"]
+            })
+    
+    return messages
+
+def generate_answer(query, chat_history=None):
+    """Genererar svar baserat på fråga, chatthistorik och retrieval-baserad kontext med Claude Sonnet 4."""
     # Hämta relevant kontext via RAG istället för hela databasen
     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
+    # System-prompts
     system_prompt = prompt_template
     
-    # Skapa ett renare användarmeddelande med bara den relevanta kontexten
+    # Förbered meddelanden för Claude API
+    messages = []
+    
+    # Lägg till chatthistorik om den finns
+    if chat_history and len(chat_history) > 1:
+        # Formatera tidigare meddelanden (exkludera det senaste som är den aktuella frågan)
+        formatted_history = format_chat_history_for_claude(chat_history[:-1])
+        messages.extend(formatted_history)
+    
+    # Skapa användarmeddelandet med kontext och aktuell fråga
     user_message = f"""Jag har en fråga om ChargeNode.
 
 Relevant kontext för frågan:
@@ -382,20 +409,22 @@ Relevant kontext för frågan:
 
 Min fråga är: {query}"""
     
+    # Lägg till det aktuella användarmeddelandet
+    messages.append({"role": "user", "content": user_message})
+    
     try:
-        # Använd Claude Haiku med RAG-baserad kontext
+        # Använd Claude Sonnet 4 med RAG-baserad kontext och chatthistorik
         response = anthropic_client.messages.create(
-            model="claude-3-7-sonnet-20250219",
+            model=MODEL_NAME,
             max_tokens=500,
             temperature=0.3,
             system=system_prompt,
-            messages=[
-                {"role": "user", "content": user_message}
-            ]
+            messages=messages
         )
         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:
+        print(f"Fel vid API-anrop: {str(e)}")
         return f"Tekniskt fel: {str(e)}\n\nAI-genererat. Kontakta support@chargenode.eu eller 010-2051055"
 
 # --- Slack Integration ---
@@ -809,7 +838,9 @@ with gr.Blocks(css=custom_css, title="ChargeNode Kundtjänst") as app:
     def respond(message, chat_history, request: gr.Request):
         global last_log
         start = time.time()
-        response = generate_answer(message)
+        
+        # Skicka hela chatthistoriken till generate_answer
+        response = generate_answer(message, chat_history)
         elapsed = round(time.time() - start, 2)
 
         timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -846,7 +877,8 @@ with gr.Blocks(css=custom_css, title="ChargeNode Kundtjänst") as app:
             "ip": ip,
             "browser": browser,
             "os": osys,
-            "platform": platform
+            "platform": platform,
+            "chat_history_length": len(chat_history)  # Lägg till information om chatthistorikens längd
         }
 
         # Använd den förbättrade loggfunktionen
@@ -863,7 +895,7 @@ with gr.Blocks(css=custom_css, title="ChargeNode Kundtjänst") as app:
 
 *Bot:* {response[:300]}{'...' if len(response) > 300 else ''}
 
-*Sessionsinfo:* {session_id[:8]}... | {browser} | {platform}
+*Sessionsinfo:* {session_id[:8]}... | {browser} | {platform} | Chat längd: {len(chat_history)} meddelanden
 """
             # Skicka asynkront för att inte blockera svarstiden
             threading.Thread(
@@ -984,7 +1016,7 @@ with gr.Blocks(css=custom_css, title="ChargeNode Kundtjänst") as app:
             }
     
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: None, None, chatbot, queue=False)
+    clear.click(lambda: initial_chat, None, chatbot, queue=False)
     support_btn.click(show_support_form, chatbot, [chat_interface, support_interface, success_interface, chat_preview])
     back_btn.click(back_to_chat, None, [chat_interface, support_interface, success_interface])
     back_to_chat_btn.click(back_to_chat, None, [chat_interface, support_interface, success_interface])