André Oliveira commited on
Commit
ec21b79
·
1 Parent(s): 9d761b8

fix: http connection not working

Browse files
Files changed (1) hide show
  1. app.py +13 -5
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://localhost:8000"
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