Update app.py

app.py

@@ -1,11 +1,11 @@
-import os
 import gradio as gr
+import os
 import requests
 from llama_cpp import Llama
 
 # Define model URL & local path
 MODEL_URL = "https://huggingface.co/TheBloke/Llama-2-7B-GGML/resolve/main/llama-7B.ggmlv3.q4_0.bin"
-MODEL_PATH = "/home/user/app/llama-7B.ggmlv3.q4_0.bin"  # Local storage path
+MODEL_PATH = "C:/Users/Prof. Mushongera/Desktop/omniAI/llama-7B.ggmlv3.q4_0.bin"  # Local storage path
 
 # Function to download model if not present
 def download_model():
@@ -20,13 +20,16 @@ def download_model():
 # Download model before launching OmniAI
 download_model()
 
-# Define OmniAI's chat function
+# Define OmniAI's chat function (handles two arguments: user message + history)
 def omni_ai_chat(user_message, history):
-    llm = Llama(model_path=MODEL_PATH)  # Load locally downloaded model
-    response = llm(user_message)
-    return response["choices"][0]["text"].strip()
+    try:
+        llm = Llama(model_path=MODEL_PATH)  # Load locally downloaded model
+        response = llm(user_message)  # Process only the latest prompt
+        return response["choices"][0]["text"].strip()
+    except Exception as e:
+        return f"Error loading AI model: {str(e)}"
 
-# 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!",
                            type="messages")  # Fixes deprecated format warning