Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -2074,12 +2074,12 @@ def load_fields(backup_json: str):
|
|
| 2074 |
logging.error(f"load_fields error: {e}")
|
| 2075 |
return ("", "", "", "", "", "", [], 10, 4, 2, "", "", "", "", "", "")
|
| 2076 |
|
| 2077 |
-
def refine_query(query: str, openai_api_key: str,
|
| 2078 |
os.environ["OPENAI_API_KEY"] = openai_api_key
|
| 2079 |
if not query:
|
| 2080 |
logging.info("refine_query: Empty query provided.")
|
| 2081 |
return ""
|
| 2082 |
-
prompt = f"Reformulate the following research query in more detailed format suitable for deep research: '{query}' \n Use this following style: {
|
| 2083 |
refined = openai_call(prompt, model="gpt-4o-mini", max_tokens_param=500)
|
| 2084 |
logging.info(f"refine_query: Refined query: {refined}")
|
| 2085 |
return refined
|
|
@@ -2805,7 +2805,9 @@ def main():
|
|
| 2805 |
with gr.Accordion ("2] Research topic", open=False):
|
| 2806 |
with gr.Row():
|
| 2807 |
research_query = gr.Textbox(label="Research Query", placeholder="Enter your research query here...", lines=2, elem_id="research-query", scale=4)
|
| 2808 |
-
|
|
|
|
|
|
|
| 2809 |
|
| 2810 |
with gr.Accordion("3] Q&A", open=False):
|
| 2811 |
with gr.Row():
|
|
@@ -2823,7 +2825,6 @@ def main():
|
|
| 2823 |
surprise_me_checkbox = gr.Checkbox(label="Surprise me", value=False)
|
| 2824 |
with gr.Column():
|
| 2825 |
additional_clarifications = gr.Textbox(label="Additional Clarifications", placeholder="Clarifications to be applied between iterations", lines=2, scale=4)
|
| 2826 |
-
reportstyle = gr.Textbox(label="Report style", placeholder="The report style", lines=1, value="Academic style", scale=1)
|
| 2827 |
selected_engines = gr.CheckboxGroup(label="Specific engines to Use (AI to choose by default)", choices=[
|
| 2828 |
"google",
|
| 2829 |
"google_jobs_listing",
|
|
@@ -2892,7 +2893,7 @@ def main():
|
|
| 2892 |
# Button actions
|
| 2893 |
refine_query_button.click(
|
| 2894 |
fn=refine_query,
|
| 2895 |
-
inputs=[research_query, openai_api_key_input,
|
| 2896 |
outputs=[research_query]
|
| 2897 |
)
|
| 2898 |
gen_followups.click(
|
|
|
|
| 2074 |
logging.error(f"load_fields error: {e}")
|
| 2075 |
return ("", "", "", "", "", "", [], 10, 4, 2, "", "", "", "", "", "")
|
| 2076 |
|
| 2077 |
+
def refine_query(query: str, openai_api_key: str, reportstyle: str) -> str:
|
| 2078 |
os.environ["OPENAI_API_KEY"] = openai_api_key
|
| 2079 |
if not query:
|
| 2080 |
logging.info("refine_query: Empty query provided.")
|
| 2081 |
return ""
|
| 2082 |
+
prompt = f"Reformulate the following research query in more detailed format suitable for deep research: '{query}' \n Use this following style: {reportstyle}"
|
| 2083 |
refined = openai_call(prompt, model="gpt-4o-mini", max_tokens_param=500)
|
| 2084 |
logging.info(f"refine_query: Refined query: {refined}")
|
| 2085 |
return refined
|
|
|
|
| 2805 |
with gr.Accordion ("2] Research topic", open=False):
|
| 2806 |
with gr.Row():
|
| 2807 |
research_query = gr.Textbox(label="Research Query", placeholder="Enter your research query here...", lines=2, elem_id="research-query", scale=4)
|
| 2808 |
+
with gr.Column(scale=1):
|
| 2809 |
+
reportstyle = gr.Textbox(label="Report style", placeholder="The report style", lines=1, value="Academic style")
|
| 2810 |
+
refine_query_button = gr.Button("Refine my Query")
|
| 2811 |
|
| 2812 |
with gr.Accordion("3] Q&A", open=False):
|
| 2813 |
with gr.Row():
|
|
|
|
| 2825 |
surprise_me_checkbox = gr.Checkbox(label="Surprise me", value=False)
|
| 2826 |
with gr.Column():
|
| 2827 |
additional_clarifications = gr.Textbox(label="Additional Clarifications", placeholder="Clarifications to be applied between iterations", lines=2, scale=4)
|
|
|
|
| 2828 |
selected_engines = gr.CheckboxGroup(label="Specific engines to Use (AI to choose by default)", choices=[
|
| 2829 |
"google",
|
| 2830 |
"google_jobs_listing",
|
|
|
|
| 2893 |
# Button actions
|
| 2894 |
refine_query_button.click(
|
| 2895 |
fn=refine_query,
|
| 2896 |
+
inputs=[research_query, openai_api_key_input, reportstyle],
|
| 2897 |
outputs=[research_query]
|
| 2898 |
)
|
| 2899 |
gen_followups.click(
|