André Oliveira
commited on
Commit
·
ec21b79
1
Parent(s):
9d761b8
fix: http connection not working
Browse files
app.py
CHANGED
|
@@ -3,9 +3,17 @@ import requests
|
|
| 3 |
import json
|
| 4 |
from models import OptimizeRequest, AutotuneRequest, QARequest
|
| 5 |
import os
|
|
|
|
|
|
|
|
|
|
| 6 |
|
| 7 |
-
API_URL = "http://
|
| 8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
# -------------------------------
|
| 11 |
# Helper to call API
|
|
@@ -79,8 +87,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
| 79 |
label="Drag & Drop Files"
|
| 80 |
)
|
| 81 |
upload_path = gr.Textbox(value=DEFAULT_UPLOAD_PATH, label="Docs Path")
|
| 82 |
-
upload_output = gr.Textbox(label="Response")
|
| 83 |
upload_btn = gr.Button("Upload",variant='primary')
|
|
|
|
| 84 |
upload_btn.click(upload_docs_tool, inputs=[upload_files, upload_path], outputs=upload_output)
|
| 85 |
gr.Markdown("---")
|
| 86 |
|
|
@@ -88,8 +96,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
| 88 |
with gr.Column():
|
| 89 |
gr.Markdown("## Optimize RAG\nRun full RAG optimization with custom parameters.")
|
| 90 |
optimize_input = gr.Textbox(lines=12, value=DEFAULT_OPTIMIZE_JSON, label="OptimizeRequest JSON")
|
| 91 |
-
optimize_output = gr.Textbox(lines=15, label="Response")
|
| 92 |
optimize_btn = gr.Button("Submit",variant='primary')
|
|
|
|
| 93 |
optimize_btn.click(optimize_rag_tool, inputs=optimize_input, outputs=optimize_output)
|
| 94 |
gr.Markdown("---")
|
| 95 |
|
|
@@ -97,8 +105,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
| 97 |
with gr.Column():
|
| 98 |
gr.Markdown("## Autotune RAG\nRun AutoRAG tuner and full optimization.")
|
| 99 |
autotune_input = gr.Textbox(lines=12, value=DEFAULT_AUTOTUNE_JSON, label="AutotuneRequest JSON")
|
| 100 |
-
autotune_output = gr.Textbox(lines=15, label="Response")
|
| 101 |
autotune_btn = gr.Button("Submit",variant='primary')
|
|
|
|
| 102 |
autotune_btn.click(autotune_tool, inputs=autotune_input, outputs=autotune_output)
|
| 103 |
gr.Markdown("---")
|
| 104 |
|
|
@@ -106,8 +114,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
| 106 |
with gr.Column():
|
| 107 |
gr.Markdown("## Generate QA\nGenerate validation QA dataset from documents.")
|
| 108 |
qa_input = gr.Textbox(lines=12, value=DEFAULT_QA_JSON, label="QARequest JSON")
|
| 109 |
-
qa_output = gr.Textbox(lines=15, label="Response")
|
| 110 |
qa_btn = gr.Button("Submit",variant='primary')
|
|
|
|
| 111 |
qa_btn.click(generate_qa_tool, inputs=qa_input, outputs=qa_output)
|
| 112 |
gr.Markdown("---")
|
| 113 |
|
|
|
|
| 3 |
import json
|
| 4 |
from models import OptimizeRequest, AutotuneRequest, QARequest
|
| 5 |
import os
|
| 6 |
+
import threading
|
| 7 |
+
import uvicorn
|
| 8 |
+
from api import app as fastapi_app
|
| 9 |
|
| 10 |
+
API_URL = "http://127.0.0.1:7861"
|
| 11 |
|
| 12 |
+
def start_fastapi():
|
| 13 |
+
uvicorn.run(fastapi_app, host="0.0.0.0", port=7861, log_level="info")
|
| 14 |
+
|
| 15 |
+
# Start FastAPI in a background thread
|
| 16 |
+
threading.Thread(target=start_fastapi, daemon=True).start()
|
| 17 |
|
| 18 |
# -------------------------------
|
| 19 |
# Helper to call API
|
|
|
|
| 87 |
label="Drag & Drop Files"
|
| 88 |
)
|
| 89 |
upload_path = gr.Textbox(value=DEFAULT_UPLOAD_PATH, label="Docs Path")
|
|
|
|
| 90 |
upload_btn = gr.Button("Upload",variant='primary')
|
| 91 |
+
upload_output = gr.Textbox(label="Response")
|
| 92 |
upload_btn.click(upload_docs_tool, inputs=[upload_files, upload_path], outputs=upload_output)
|
| 93 |
gr.Markdown("---")
|
| 94 |
|
|
|
|
| 96 |
with gr.Column():
|
| 97 |
gr.Markdown("## Optimize RAG\nRun full RAG optimization with custom parameters.")
|
| 98 |
optimize_input = gr.Textbox(lines=12, value=DEFAULT_OPTIMIZE_JSON, label="OptimizeRequest JSON")
|
|
|
|
| 99 |
optimize_btn = gr.Button("Submit",variant='primary')
|
| 100 |
+
optimize_output = gr.Textbox(lines=15, label="Response")
|
| 101 |
optimize_btn.click(optimize_rag_tool, inputs=optimize_input, outputs=optimize_output)
|
| 102 |
gr.Markdown("---")
|
| 103 |
|
|
|
|
| 105 |
with gr.Column():
|
| 106 |
gr.Markdown("## Autotune RAG\nRun AutoRAG tuner and full optimization.")
|
| 107 |
autotune_input = gr.Textbox(lines=12, value=DEFAULT_AUTOTUNE_JSON, label="AutotuneRequest JSON")
|
|
|
|
| 108 |
autotune_btn = gr.Button("Submit",variant='primary')
|
| 109 |
+
autotune_output = gr.Textbox(lines=15, label="Response")
|
| 110 |
autotune_btn.click(autotune_tool, inputs=autotune_input, outputs=autotune_output)
|
| 111 |
gr.Markdown("---")
|
| 112 |
|
|
|
|
| 114 |
with gr.Column():
|
| 115 |
gr.Markdown("## Generate QA\nGenerate validation QA dataset from documents.")
|
| 116 |
qa_input = gr.Textbox(lines=12, value=DEFAULT_QA_JSON, label="QARequest JSON")
|
|
|
|
| 117 |
qa_btn = gr.Button("Submit",variant='primary')
|
| 118 |
+
qa_output = gr.Textbox(lines=15, label="Response")
|
| 119 |
qa_btn.click(generate_qa_tool, inputs=qa_input, outputs=qa_output)
|
| 120 |
gr.Markdown("---")
|
| 121 |
|