Update app.py

app.py

@@ -106,14 +106,15 @@ 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(response)
-        reponame=""
-        filename=""
-        filecontent=""
+        yield reponame='test1'
+        yield filename=data[1]
+        yield filecontent=response
         
-        yield output
-        yield history
-        yield prompt
+        #yield output
+        #yield history
+        #yield prompt
     #with open(f'{loc_folder}/{loc_file}.json','w') as jobj:
     #    lod.append({'prompt':prompt,'response':output,'image':im_box,'model':clients[1]['name'],'seed':seed}),
     #    jobj.write(json.dumps(lod,indent=4))
@@ -197,7 +198,8 @@ def agent(prompt_in,history,mod=2):
                     fn="NONE"
                 if 'CREATE_FILE' in fn:
                     print('CREATE_FILE called')
-                    out_w =generate(com,history,mod=mod,tok=10000,seed=seed,role="CREATE_FILE",data=in_data)
+                    in_data[1]=com
+                    out_w =generate(prompt,history,mod=mod,tok=10000,seed=seed,role="CREATE_FILE",data=in_data)
                     build_space(out_w[0],out_w[1],out_w[2])
                 elif 'IMAGE' in fn:
                     print('IMAGE called')