Coder19/interview_system

1from fastapi import APIRouter2from pydantic import BaseModel3import openai4from fastapi import APIRouter, HTTPException5from peft import PeftModel, PeftConfig6from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig7import torch8from interview_prep.Text_generation import generate_output9router = APIRouter()10 11# 12class QuestionRequest(BaseModel):13    domain: str14# 15class QuestionRequest(BaseModel):16    prompt: str  # Accepts the prompt from the user17 18@router.post("/generate_question")19def generate_question(request: QuestionRequest):20    try:21        generated_text = generate_output(request.prompt)22        return {"generated_text": generated_text}23    except Exception as e:24        raise HTTPException(status_code=500, detail=str(e))25