lalashechka/open_api

1from fastapi import FastAPI, Request2from pydantic import BaseModel3import requests4import os5import uuid6import re7 8 9app = FastAPI()10 11 12class RequestBody(BaseModel):13    model: str14    text: str15 16 17@app.post("/api/v1")18async def generate_response(request_body: RequestBody):19    input_text = request_body.text20    model = request_body.model21    if model == 'blackbox':22        headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36","Accept": "*/*","Accept-Language": "en-US,en;q=0.5","Accept-Encoding": "gzip, deflate, br","Referer": "https://www.blackbox.ai","Content-Type": "application/json","Origin": "https://www.blackbox.ai","DNT": "1","Sec-GPC": "1","Alt-Used": "www.blackbox.ai","Connection": "keep-alive",}23        json_data = {'messages': [{'id':  uuid.uuid4().hex,'content': input_text,'role': 'user',},],'id':  uuid.uuid4().hex,'previewToken': None,'userId': None,'codeModelMode': True,'agentMode': { 'mode': True, 'id': 'ImageGenerationLV45LJp', 'name': 'Image Generation',},'trendingAgentMode': {},'isMicMode': False,'maxTokens': 1024,'playgroundTopP': None,'playgroundTemperature': None,'isChromeExt': False,'githubToken': None,'clickedAnswer2': False,'clickedAnswer3': False,'clickedForceWebSearch': False,'visitFromDelta': False,'mobileClient': False,'userSelectedModel': None,'validated': '00f37b34-a166-4efb-bce5-1312d87f2f94',}24        response = requests.post('https://www.blackbox.ai/api/chat', headers=headers, json=json_data)25        decoded_string = ''.join(char for char in response.text if char != '\x00' and char != '\x08')26        pattern = r'https?://\S+\.(?:jpg|jpeg|png|gif)'27        match2 = re.search(pattern, decoded_string)28        all_chunk = match2.group()29        30    return {"response": all_chunk}