import os
import logging
import pandas as pd
import streamlit as st
import requests
from transformers import pipeline

# Set up logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Load environment variables from .env file
load_dotenv()

# Get the Hugging Face API key from environment variables
hf_api_key = os.getenv("HUGGINGFACE_API_KEY")
if not hf_api_key:
    raise ValueError("HUGGINGFACE_API_KEY is not set. Please provide a valid API key.")

# Hugging Face API URL
hf_api_url = "https://api-inference.huggingface.co/models/{model_name}"

# Function to load and preprocess data
@st.cache_data
def load_data(file):
    try:
        df = pd.read_csv(file)
        return df
    except Exception as e:
        logger.error(f"Error loading CSV file: {e}")
        st.error("There was an issue loading the file. Please try again.")
        return pd.DataFrame()  # Return an empty DataFrame in case of error

# Function to call Hugging Face API for text generation
def generate_text_from_model(model_name, text_input):
    headers = {"Authorization": f"Bearer {hf_api_key}"}
    data = {"inputs": text_input}

    try:
        response = requests.post(hf_api_url.format(model_name=model_name), headers=headers, json=data)
        response.raise_for_status()
        result = response.json()
        if 'generated_text' in result:
            return result['generated_text']
        else:
            return "No result from model. Please try again."
    except requests.exceptions.RequestException as err:
        logger.error(f"Error interacting with Hugging Face API: {err}")
        st.error(f"Error interacting with Hugging Face API: {err}")
        return ""

# Streamlit app layout
def main():
    # Set a background color and style
    st.markdown(
        """
        <style>
        .stApp {
            background-color: #F4F4F9;
        }
        .stButton>button {
            background-color: #6200EE;
            color: white;
            font-size: 18px;
        }
        .stSlider>div>div>span {
            color: #6200EE;
        }
        .stTextInput>div>div>input {
            background-color: #E0E0E0;
        }
        </style>
        """,
        unsafe_allow_html=True
    )

    # Title and header
    st.title("🌟 **Hugging Face Text Generation** 🌟")
    st.markdown("### **Generate text using Hugging Face Models**")

    # User input for text generation
    model_name = st.selectbox("🔹 Select Hugging Face Model", ["gpt2", "distilgpt2", "t5-small"])
    text_input = st.text_area("🔹 Input Text", "Once upon a time...")

    # Generate text based on input
    if st.button("🔍 Generate Text"):
        st.subheader("🔔 **Generated Text** 🔔")
        generated_text = generate_text_from_model(model_name, text_input)
        st.write(f"📜 {generated_text}")

if __name__ == "__main__":
    main()