app.py

import os
import subprocess
import gradio as gr
import ollama

# Ensure ollama is installed via pip
def install_ollama():
    try:
        import ollama
        print("✅ Ollama is already installed.")
    except ImportError:
        print("🚀 Installing Ollama via pip...")
        subprocess.run(["pip", "install", "ollama"], check=True)
        import ollama  # Re-import after installation
        print("✅ Ollama installed successfully!")

# Ensure model is downloaded
MODEL_NAME = "deepseek-llm-7b"
MODEL_PATH = f"models/{MODEL_NAME}"

def download_model():
    models = ollama.list()
    if any(m["name"] == MODEL_NAME for m in models["models"]):
        print(f"✅ Model '{MODEL_NAME}' is already available.")
    else:
        print(f"🚀 Downloading model: {MODEL_NAME} ...")
        ollama.pull(MODEL_NAME)
        print(f"✅ Model '{MODEL_NAME}' downloaded successfully.")

# Generate AI response using Ollama
def chat_response(user_input):
    response = ollama.chat(model=MODEL_NAME, messages=[{"role": "user", "content": user_input}])
    return response['message']['content']

# Run setup
install_ollama()
download_model()

# Create Gradio Interface
iface = gr.Interface(
    fn=chat_response,
    inputs="text",
    outputs="text",
    title="DeepSeek ChatBot (Ollama)",
    description="Chat with DeepSeek LLM 7B using Ollama."
)

# Launch Gradio App
if __name__ == "__main__":
    iface.launch(server_name="0.0.0.0", server_port=7860)