Update app.py

app.py

@@ -40,14 +40,14 @@ def predict(text, url_params):
     if load_game != None:
         try:
             
-            return "" + text + ""
+            return "" + text + "", "True"
         except Exception as e:
             print(f"cannot load game.  Error::{e}")
             pass
-        
     else:
        pass
-    return ["" + text + ""]
+    return ["" + text + "", "False"]
+    
 def parse_action(string: str):
     print("PARSING:")
     print(string)
@@ -338,44 +338,46 @@ def run_action(purpose, task, history, action_name, action_input,result):
 
         return "MAIN", None, history, task, result
 
-def run(purpose,history):
-    yield "Searching..."
-    task=None
-    result=""
-    #history = ""
-    if not history:
-        history = ""
-    else:
-        history=str(history)
-    action_name = "MAIN"
-    action_input = None
-    while True:
-        print("")
-        print("")
-        print("---")
-        #print("purpose:", purpose)
-        print("task:", task)
-        print("---")
-        #print(history)
-        print("---")
-
-        action_name, action_input, history, task, result = run_action(
-            purpose,
-            task,
-            history,
-            action_name,
-            action_input,
-            result
-        )
-            
-        if not result:
-            yield "More Searching..."
+def run(purpose,history,key_inp):
+    if key_inp=="True":    
+        yield "Searching..."
+        task=None
+        result=""
+        #history = ""
+        if not history:
+            history = ""
         else:
-            yield result
-        if action_name == "COMPLETE":
-            break
-    return result
-
+            history=str(history)
+        action_name = "MAIN"
+        action_input = None
+        while True:
+            print("")
+            print("")
+            print("---")
+            #print("purpose:", purpose)
+            print("task:", task)
+            print("---")
+            #print(history)
+            print("---")
+    
+            action_name, action_input, history, task, result = run_action(
+                purpose,
+                task,
+                history,
+                action_name,
+                action_input,
+                result
+            )
+                
+            if not result:
+                yield "More Searching..."
+            else:
+                yield result
+            if action_name == "COMPLETE":
+                break
+        return result
+    else:
+        return "Invalid Key"
 
 
 
@@ -386,10 +388,13 @@ examples =[
     "Teach me to fly a helicopter"
 ]
 with gr.Blocks() as app:
+    key_inp=gr.Textbox()
+    add_in=[key_inp]
     gr.ChatInterface(
         fn=run,
-        chatbot=gr.Chatbot(show_label=False, show_share_button=True, show_copy_button=True, layout="panel", height="500px"),
         title="<center>Mixtral 46.7B Powered <br>Financial Demo<br>This demo is not a Financial Advisor</center>",
+        chatbot=gr.Chatbot(show_label=False, show_share_button=True, show_copy_button=True, layout="panel", height="500px"),
+        additional_inputs=add_in,
         examples=examples,
         concurrency_limit=20,
     )