Update app.py

app.py

@@ -106,6 +106,7 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
         stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
         for response in stream:
             output += response.token.text
+        
         #print(file_content)
         print(output)
         yield 'test1'
@@ -129,7 +130,34 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
     image_out.save(output)
     print('Done: ',output)
     return [{'role':'assistant','content': {'path':output}}]'''
-
+def parse_json(inp):
+    if type(inp)==type(""):
+        if "```" in inp:
+            tog=False
+            cnt=0
+            lines=""
+            for line in inp.split("\n"):
+                if "```" in line:
+                    cnt+=1
+                    if cnt==1:
+                        tog = True
+                    elif cnt==2:
+                        tog = False
+                if tog==True:
+                   lines+=line
+        try:
+            print(lines)
+            out_json=eval(lines)
+            out1=out_json['filename']
+            out2=out_json['filecontent']
+            return out1,out2
+        except Exception as e:
+            print(e)
+            return "None","None"
+    elif type(inp)==type({}):
+        out1=inp['filename']
+        out2=inp['filecontent']
+        return out1,out2
 def build_space(repo_name,file_name,file_content,access_token=""):
     try:
         repo_path="broadfield/"+repo_name
@@ -143,6 +171,7 @@ def build_space(repo_name,file_name,file_content,access_token=""):
                 exist_ok=True,
                 private=False,
             )
+        
         # Create a new Space
         #response = api.create_repo(repo_path)
         #space_info = repo_url.json()
@@ -212,6 +241,7 @@ def agent(prompt_in,history,mod=2):
                     in_data[1]=com
                     out_o =generate(prompt,history,mod=mod,tok=10000,seed=seed,role="CREATE_FILE",data=in_data)
                     out_w=list(out_o)
+                    ret1,ret2 = parse_json(out_w[2])
                     build_space(out_w[0],out_w[1],out_w[2])
                 elif 'IMAGE' in fn:
                     print('IMAGE called')
@@ -226,8 +256,8 @@ def agent(prompt_in,history,mod=2):
                 elif 'NONE' in fn:
                     print('ERROR ACTION NOT FOUND')
                     history+=[{'role':'system','content':f'observation:The last thing we attempted resulted in an error, check formatting on the tool call'}]
-
                 else:pass;seed = random.randint(1,9999999999999)
+                    
 with gr.Blocks() as ux:
     with gr.Row():
         with gr.Column():