Spaces:
Sleeping
Sleeping
File size: 1,047 Bytes
89dc8b2 e960d63 89dc8b2 e960d63 89dc8b2 e960d63 89dc8b2 |
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 |
from fastapi import FastAPI
from pydantic import BaseModel
import faq as faq
import uvicorn
import gradio as gr
app = FastAPI()
class Request(BaseModel):
question: str
sheet_url: str
page_content_column: str
k: int
@app.post("/api/v1/ask")
async def ask_api(request: Request):
return ask(
request.sheet_url, request.page_content_column, request.k, request.question
)
def ask(sheet_url: str, page_content_column: str, k: int, question: str):
vectordb = faq.load_vectordb(sheet_url, page_content_column)
result = faq.similarity_search(vectordb, question, k=k)
return result
iface = gr.Interface(
fn=ask,
inputs=[
gr.Textbox(label="Google Sheet URL"),
gr.Textbox(label="Question Column"),
gr.Slider(1, 5, step=1, label="K"),
gr.Textbox(label="Question"),
],
outputs=[gr.JSON(label="Answer")],
allow_flagging="never",
)
app = gr.mount_gradio_app(app, iface, path="/")
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=7860)
|