Guiyom commited on
Commit
78bba40
·
verified ·
1 Parent(s): 2e33ae1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
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, style: 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: {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
- refine_query_button = gr.Button("Refine my Query", scale=1)
 
 
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, reportsyle],
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(