Update app.py

app.py

@@ -1,17 +1,23 @@
 """SusAI ©2025 Intern Labs. v1.1.0"""
 import os
 import gradio as gr
-
 from gradio_client import Client
 
 # Initialize Hugging Face Inference Client
-client = Client("HydroFlyer53/ThePickle", hf_token=os.environ.get("Key"))
+client = Client("HydroFlyer53/ThePickle", hf_token=os.environ["Key"])
 
 def chat_with_ai(message, history):
-    """Function to get AI response from Hugging Face model."""
+    """Function to get AI response from Hugging Face model with limited history tracking."""
+
+    # Retain only the last 3 exchanges in history
+    trimmed_history = history[-3:] if history else []
+    formatted_history = "\n".join([f"User: {h[0]}\nAI: {h[1]}" for h in trimmed_history])
+
+    # Combine history with new message
+    prompt = f"{formatted_history}\nUser: {message}\nAI:"
 
     result = client.predict(
-        message=message,
+        message=prompt,
         system_message=(
             "You are an AI that talks in Gen-Z slang, and also says things like skibbidy and sigma, "
             "but aren't really that smart or helpful. If you are asked to stop talking in slang, "
@@ -24,10 +30,11 @@ def chat_with_ai(message, history):
         top_p=0.60,
         api_name="/chat"
     )
+
     return result
 
-# Gradio Chat Interface
+# Gradio Chat Interface with history
 demo = gr.ChatInterface(fn=chat_with_ai)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()