import streamlit as st
import requests
import os

# Get Together API key from environment variable
TOGETHER_API_KEY = os.getenv('TOGETHER_API_KEY')

# Streamlit UI
st.title("Movie Review Sentiment Analysis")

# User input for movie review
zero_shot_prompt = st.text_area("Enter zero-shot prompt:", "")
one_shot_prompt = st.text_area("Enter one-shot prompt:", "")
few_shot_prompt = st.text_area("Enter few-shot prompts:", "")

# Together API endpoint
endpoint = 'https://api.together.xyz/inference' 

# Determine the prompt based on user input
if st.button("Analyze Zero-shot"):
    prompt = zero_shot_prompt
elif st.button("Analyze One-shot"):
    prompt = one_shot_prompt
elif st.button("Analyze Few-shot"):
    prompt = few_shot_prompt
else:
    prompt = ""

# Make a request to Together API if a prompt is provided
if prompt:
    response = requests.post(endpoint, json={
        "model": 'togethercomputer/RedPajama-INCITE-7B-Base',
        "prompt": prompt,
        "top_p": 1,
        "top_k": 40, 
        "temperature": 0.8,
        "max_tokens": 1,
        "repetition_penalty": 1,
    }, headers={
        "Authorization": f"Bearer {TOGETHER_API_KEY}"
    })

    # Display the output
    st.text("Sentiment Analysis Result:")
    st.text(response.json()['output']['choices'][0]['text'])