# CodeGuru GenAI App

import os
import gradio as gr
import requests
import json

HF_TOKEN = os.environ.get("HF_TOKEN")
API_URL = "https://api-inference.huggingface.co/models/Salesforce/codegen-2B-multi"

headers = {
    "Authorization": f"Bearer {HF_TOKEN}",
    "Content-Type": "application/json"
}

history = []

def generate_response(prompt):
    history.append(prompt)
    final_prompt = "\n".join(history)

    payload = {
        "inputs": final_prompt,
        "parameters": {
            "temperature": 1
        }
    }

    response = requests.post(API_URL, headers=headers, data=json.dumps(payload))

    if response.status_code == 200:
        result = response.json()
        return result[0]["generated_text"] if isinstance(result, list) else result.get("generated_text", "No response.")
    else:
        return f"Error: {response.status_code} - {response.text}"

interface = gr.Interface(
    fn=generate_response,
    inputs=gr.Textbox(lines=4, placeholder="Ask CodeGuru your coding question..."),
    outputs="text",
    title="CodeGuru - Your Code Assistant"
)

interface.launch()