disable tmp key

app.py

@@ -1,18 +1,11 @@
 import gradio as gr
 import os
-with open(os.path.join('./data/tmp_key/', 'openai.key'), 'r') as f:
-    os.environ["OPENAI_API_KEY"] = f.read().strip()
-with open(os.path.join('./data/tmp_key/', 'serpapi.key'), 'r') as f:
-    os.environ["SERPAPI_API_KEY"] = f.read().strip()
 
-from algos.PWS import *
-from utils.util import *
-
-def process(tools, model, input_text):
-    # This is a placeholder function. Replace it with your own logic.
-    planner = "Planner output for " + input_text
-    solver = "Solver output for " + input_text
-    output = "Output for " + input_text
+def process(tools, model, input_text, openai_key, serpapi_key):
+    os.environ["OPENAI_API_KEY"] = openai_key
+    os.environ["SERPAPI_API_KEY"] = serpapi_key
+    from algos.PWS import PWS_Base
+    from utils.util import fewshots
 
     method = PWS_Base(planner_model=model, solver_model=model,
                   fewshot=fewshots.TRIVIAQA_PWS, available_tools=tools)
@@ -25,13 +18,15 @@ def process(tools, model, input_text):
 tools = gr.components.CheckboxGroup(['Wikipedia', 'Google', 'LLM', 'WolframAlpha', 'Calculator'],label="Tools")
 model = gr.components.Dropdown(["text-davinci-003", "gpt-3.5-turbo"], label="Model")
 input_text = gr.components.Textbox(lines=2, placeholder="Input Here...", label="Input")
+openai_key = gr.components.Textbox(lines=1, placeholder="Provide your openai key Here...", label="OpenAI Key")
+serpapi_key = gr.components.Textbox(lines=1, placeholder="Only needed if you use Google tool.", label="SerpAPI Key")
 planner = gr.components.Textbox(lines=4, label="Planner")
 solver = gr.components.Textbox(lines=4, label="Solver")
 output = gr.components.Textbox(label="Output")
 
 iface = gr.Interface(
     fn=process,
-    inputs=[tools, model, input_text],
+    inputs=[tools, model, input_text, openai_key, serpapi_key],
     outputs=[planner, solver, output],
     examples=[
         [["Wikipedia", "LLM"], "gpt-3.5-turbo", "American Callan Pinckney’s eponymously named system became a best-selling (1980s-2000s) book/video franchise in what genre?"],

data/tmp_key/openai.key

@@ -1 +0,0 @@
-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

data/tmp_key/serpapi.key

@@ -1 +0,0 @@
-8a158c526d6f258efab645fc935e846655678766f05ec444aa19009eac63989e