Update app.py

app.py

@@ -1,18 +1,19 @@
 import gradio as gr
 from llama_cpp import Llama
 
-# Load the model directly from Hugging Face's cloud
+# Load the model directly from Hugging Face’s cloud
 MODEL_URL = "https://huggingface.co/TheBloke/Llama-2-7B-GGML/resolve/main/llama-7B.ggmlv3.q4_0.bin"
 
-# Define OmniAI's chat function
-def omni_ai_chat(user_input):
+# Define OmniAI's chat function (handles two arguments: user message + history)
+def omni_ai_chat(user_message, history):
     llm = Llama(model_path=MODEL_URL)  # Load model dynamically
-    response = llm(user_input)
+    response = llm(user_message)  # Process only the latest prompt
     return response["choices"][0]["text"].strip()
 
-# Set up Gradio chatbot UI
+# Set up Gradio chatbot UI with correct formatting
 chatbot = gr.ChatInterface(fn=omni_ai_chat, title="OmniAI - Cloud AI",
-                           description="Your personal AI assistant, running entirely in the cloud!")
+                           description="Your personal AI assistant, running entirely in the cloud!",
+                           type="messages")  # Fixes deprecated format warning
 
 # Launch the app!
 chatbot.launch()