Update app.py

app.py

@@ -144,8 +144,9 @@ def parse_json(inp):
 def build_space(repo_name,file_name,file_content,access_token=""):
     try:
         repo_path=user_+str(repo_)
-        if not access_token:access_token=os.environ['HF_TOKEN']
-
+        if not access_token:
+            #access_token=os.environ['HF_TOKEN']
+            return [{'role':'assistant','content': 'ENTER A HUGGINGFACE TOKEN'}]
         api=HfApi(endpoint="https://huggingface.co", token=access_token)
         repo_url = api.create_repo(
                 repo_id=repo_path,
@@ -238,6 +239,10 @@ def agent(prompt_in,history,mod=2,tok_in=""):
                     print('ret1',ret1)
                     print('ret2',ret2)
                     build_out = build_space(repo_,ret1,ret2,access_token=tok_in)
+                    if build_out[0].content=="ENTER A HUGGINGFACE TOKEN":
+                        yield [{'role':'assistant','content':"ENTER A HUGGINGFACE TOKEN"}]
+                        go=False
+                        break
                     history+=[{'role':'system','content':f'observation:{build_out}'}]
                     yield history