import gradio as gr
import requests
import json

def generate_code(prompt):
    # Use Hugging Face's free inference API
    API_URL = "https://api-inference.huggingface.co/models/microsoft/CodeGPT-small-py"
    
    headers = {
        "Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"  # You'll need to get this
    }
    
    payload = {
        "inputs": prompt,
        "parameters": {
            "max_length": 100,
            "temperature": 0.7
        }
    }
    
    try:
        response = requests.post(API_URL, headers=headers, json=payload)
        if response.status_code == 200:
            result = response.json()
            return result[0]['generated_text']
        else:
            return f"# Error: {response.status_code}\n# Try again in a moment"
    except:
        return f"# Generated template for: {prompt}\ndef your_function():\n    # Add implementation here\n    pass"

demo = gr.Interface(
    fn=generate_code,
    inputs=gr.Textbox(label="Code prompt"),
    outputs=gr.Textbox(label="Generated Code"),
    title="Code Generator",
    description="Generate code using Hugging Face API"
)

demo.launch()