Update app.py

app.py

@@ -1,29 +1,17 @@
 """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["Key"])
+client = Client("HydroFlyer53/ThePickle", hf_token=os.environ.get("Key"))
 
 def chat_with_ai(message, history):
-    """Function to get AI response from Hugging Face model with limited history tracking."""
-
-    # Retain only the last 3 exchanges in history and ensure proper formatting
-    trimmed_history = history[-3:] if history else []
-    formatted_history = "\n".join([f"User: {h[0]}\nAI: {h[1]}" for h in trimmed_history if h])
-    
-    # Ensure the history is not empty or corrupted
-    if not formatted_history.strip():
-        formatted_history = "User: Hi\nAI: What's up?"
-
-    # Combine history with new message
-    prompt = f"{formatted_history}\nUser: {message}\nAI:"
-    
-    print("DEBUG: Prompt sent to AI:", prompt)  # Debugging line to check input
+    """Function to get AI response from Hugging Face model."""
 
     result = client.predict(
-        message=prompt,
+        message=message,
         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, "
@@ -36,15 +24,10 @@ def chat_with_ai(message, history):
         top_p=0.60,
         api_name="/chat"
     )
-    
-    # Prevent endless repetition
-    if result.strip().count("Sus AI:") > 3:
-        return "Yo, I think I glitched out—let’s try that again!"
-    
     return result
 
-# Gradio Chat Interface with history
+# Gradio Chat Interface
 demo = gr.ChatInterface(fn=chat_with_ai)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()