Update app.py

app.py

@@ -63,7 +63,7 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
         system_prompt = prompts.PATH_MAKER.replace("**CURRENT_OR_NONE**",str(data[4])).replace("**PROMPT**",json.dumps(data[0],indent=4)).replace("**HISTORY**",str(history))
         formatted_prompt = format_prompt(prompt, mod, system_prompt)
     elif role == "CREATE_FILE":
-        system_prompt = prompts.CREATE_FILE.replace("**TIMELINE**",data[4]).replace("**FILENAME**",str(data[4]))
+        system_prompt = prompts.CREATE_FILE.replace("**TIMELINE**",data[4]).replace("**FILENAME**",str(data[1]))
         formatted_prompt = format_prompt(prompt, mod, system_prompt)
     elif role == "SEARCH":
         system_prompt = prompts.SEARCH.replace("**DATA**",data)
@@ -107,10 +107,10 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
         for response in stream:
             output += response.token.text
         #print(file_content)
-        print(response)
+        print(output)
         yield 'test1'
         yield data[1]
-        yield response
+        yield output
         
         #yield output
         #yield history