Update app.py

app.py

@@ -1,10 +1,20 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import re
-from chat_history import ChatHistory
+import json
+from typing import List, Tuple
 
-# Initialize chat history
-chat_manager = ChatHistory()
+# Simple chat history functions
+def load_history(filename="conversation_history.json"):
+    try:
+        with open(filename, "r") as f:
+            return json.load(f)
+    except FileNotFoundError:
+        return []
+
+def save_history(history, filename="conversation_history.json"):
+    with open(filename, "w") as f:
+        json.dump(history, f)
 
 # Initialize the model client
 client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
@@ -39,7 +49,7 @@ def respond(
     try:
         for token in client.chat_completion(
             messages,
-            max_new_tokens=max_tokens,
+            max_tokens=max_tokens,
             temperature=temperature,
             top_p=top_p,
             stream=True,
@@ -54,7 +64,11 @@ def respond(
                 yield response
         
         # Save the conversation after completion
-        chat_manager.add_message(message, response)
+        chat_history = load_history()
+        chat_history.append((message, response))
+        if len(chat_history) > 100:  # Keep last 100 messages
+            chat_history = chat_history[-100:]
+        save_history(chat_history)
         
     except Exception as e:
         yield f"I apologize, but I encountered an error: {str(e)}"