import gradio as gr
import os
import requests

# Set your Groq API Key
GROQ_API_KEY = os.getenv("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")  # or hardcode temporarily
MODEL_NAME = "llama3-70b-8192"  # You can also use "mixtral-8x7b-32768" or "gemma-7b-it"

# Define task prompts
TASK_TEMPLATES = {
    "Generate Content": "Write a detailed and engaging piece of content about the following topic:\n\n{input}",
    "Paraphrase": "Paraphrase the following text in a clear and concise way:\n\n{input}",
    "Summarize": "Summarize the following content in a few key points:\n\n{input}",
    "Correct Grammar": "Fix grammatical mistakes and rewrite this text properly:\n\n{input}",
    "Change Tone": "Rewrite the following text in a {tone} tone:\n\n{input}"
}

def query_groq_model(prompt, temperature=0.7):
    """Send request to Groq API."""
    url = "https://api.groq.com/openai/v1/chat/completions"
    headers = {
        "Authorization": "Bearer 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",
        "Content-Type": "application/json"
    }
    payload = {
        "model": MODEL_NAME,
        "messages": [{"role": "user", "content": prompt}],
        "temperature": temperature,
    }

    response = requests.post(url, headers=headers, json=payload)
    result = response.json()
    return result['choices'][0]['message']['content']

def process_text(task, user_input, tone):
    if not user_input.strip():
        return "❗ Please enter some text."

    if task == "Change Tone":
        prompt = TASK_TEMPLATES[task].format(input=user_input, tone=tone)
    else:
        prompt = TASK_TEMPLATES[task].format(input=user_input)

    try:
        output = query_groq_model(prompt)
    except Exception as e:
        output = f"⚠️ Error: {str(e)}"
    return output

# Gradio UI
with gr.Blocks(title="Content Creation & Paraphrasing Tool") as demo:
    gr.Markdown("## ✍️ Content Creation & Paraphrasing Tool (Open Source + Groq API)")
    
    with gr.Row():
        task = gr.Dropdown(
            label="Select Task",
            choices=list(TASK_TEMPLATES.keys()),
            value="Generate Content"
        )
        tone = gr.Dropdown(
            label="Tone (if applicable)",
            choices=["formal", "casual", "professional", "friendly"],
            visible=False
        )

    user_input = gr.Textbox(label="Input Text or Topic", lines=6, placeholder="Enter your text here...")
    output = gr.Textbox(label="Output", lines=8)

    task.change(lambda t: gr.update(visible=(t == "Change Tone")), inputs=task, outputs=tone)

    run_btn = gr.Button("Submit")
    run_btn.click(fn=process_text, inputs=[task, user_input, tone], outputs=output)

    gr.Markdown("Made with ❤️ using Open Source Models + Groq + Gradio")

if __name__ == "__main__":
    demo.launch()