Mindflow_server/main.py at main · UtkarshShah0/Mindflow_server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
import os
import requests
import re
from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware
from dotenv import load_dotenv

load_dotenv()

session = requests.Session()
session.post('https://cloud.mindsdb.com/cloud/login', json={
    'email': os.environ.get("USER_ID"),
    'password': os.environ.get("PASSWORD")
})


app = FastAPI()

# origins = [
#     "http://localhost",
#     "http://localhost:8000",
#     "chrome-extension://pgigegiebniooimfefalcnmnkhlndmcb"  # Add your extension's origin
# ]

origins = ["*"]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=False,
    allow_methods=["*"],
    allow_headers=["*"],
    expose_headers=["*"]
)
# app.add_middleware(
#     CORSMiddleware,
#     allow_origins=origins,
#     allow_credentials=True,
#     allow_methods=["*"],
#     allow_headers=["*"],
# )

@app.post("/chat")
async def chat(request: Request):
    form_data = await request.json()

    input = form_data["question"]
    text = f'"{input}"'

    resp = session.post('https://cloud.mindsdb.com/api/sql/query', json={'query':
        f'SELECT response from mindsdb.gpt_model WHERE author_username = "someuser" AND text = {text};'})

    return {'response' : resp.json()["data"][0][0]}


@app.post("/context")
async def summary(request: Request):
    form_data = await request.json()

    input = form_data["question"]
    question = f'"{input}"'

    context = form_data["context"]
    data  = f'"{context}"'

    resp = session.post('https://cloud.mindsdb.com/api/sql/query', json={'query':
                    f'SELECT * FROM que WHERE question = {question} AND text = {data} ;'})

    return (resp.json()["data"][0][0])


@app.post("/image")
async def image(request: Request):
    form_data = await request.json()

    base = form_data["base"]
    input = f'"{base}"'

    # extra = form_data["extra"]
    # output = f'"{extra}"'

    resp = session.post('https://cloud.mindsdb.com/api/sql/query', json={'query':
                    f'SELECT response from mindsdb.gpt_image_3 WHERE input = {input};'})

    return (resp.json()["data"][0][0])


@app.post("/web")
async def summary(request: Request):
    form_data = await request.json()

    input = form_data["question"]
    question = f'"{input}"'

    context = form_data["context"]
    url  = f'"{context}"'

    # url = '"https://docs.mindsdb.com/what-is-mindsdb"'
    resp_4 = session.post('https://cloud.mindsdb.com/api/sql/query', json={'query':
                    f'SELECT text_content FROM my_web.crawler WHERE url = {url} LIMIT 1;'})

    s = (resp_4.json()["data"][0][0])
    a = re.sub(r"\s+", "", s)
    b = a[:1500]

    print("**"*10)
    print(b)
    # question = '"what is mindsdb"'
    data = f'"{b}"'

    resp_5 = session.post('https://cloud.mindsdb.com/api/sql/query', json={'query':
                        f'SELECT * FROM que WHERE question = {question} AND text = {data} ;'})


    print(resp_5.json()["data"][0][0])

    return (resp_5.json()["data"][0][0])