import streamlit as st
import requests
import json
import os

API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
API_KEY = os.environ.get("API_KEY")
HEADERS = {"Authorization": f"Bearer {API_KEY}"}

st.set_page_config(page_title="Mistral Chatbot", layout="centered")
st.title("🤖 Mistral Chatbot")
st.markdown("An interactive chatbot powered by the Mistral-7B-Instruct-v0.3 model via Hugging Face API.")

user_input = st.text_input("You:", placeholder="Type your message here...")

if user_input:
    payload = {
        "inputs": user_input,
        "parameters": {"max_new_tokens": 150, "temperature": 0.1}
    }
    
    try:
        response = requests.post(API_URL, headers=HEADERS, json=payload)

        if response.status_code == 200:
            bot_response = response.json()[0]["generated_text"]
            st.write(f"**Bot:** {bot_response}")
        else:
            st.error(f"Error {response.status_code}: {response.json().get('error', 'Something went wrong.')}")
    except Exception as e:
        st.error(f"Error: {str(e)}")