CodeMentorAI / backend.py
nandunelapatla's picture
Upload backend.py
7898606 verified
raw
history blame contribute delete
801 Bytes
import requests
API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
headers = {"Authorization": f"Bearer your_token_here"} # Replace with your actual token or use os.getenv
def explain_code(code):
prompt = f"Explain this Python code in simple terms:\n{code}"
res = requests.post(API_URL, headers=headers, json={"inputs": prompt})
return res.json()[0]["generated_text"]
def debug_code(code):
prompt = f"Find bugs in this code:\n{code}"
res = requests.post(API_URL, headers=headers, json={"inputs": prompt})
return res.json()[0]["generated_text"]
def ask_question(q):
prompt = f"Answer this Python question:\n{q}"
res = requests.post(API_URL, headers=headers, json={"inputs": prompt})
return res.json()[0]["generated_text"]