bug fix.

auto_backgrounds.py

@@ -2,6 +2,7 @@ import os.path
 import json
 from utils.references import References
 from utils.file_operations import hash_name, make_archive, copy_templates
+from utils.tex_processing import create_copies
 from section_generator import section_generation_bg, keywords_generation, figures_generation, section_generation
 import logging
 import time
@@ -128,7 +129,8 @@ def generate_draft(title, description="", template="ICLR2022",
                 logging.info(message)
                 attempts_count += 1
                 time.sleep(20)
-
+    # post-processing
+    create_copies(destination_folder)
     input_dict = {"title": title, "description": description, "generator": "generate_draft"}
     filename = hash_name(input_dict) + ".zip"
     print("\nMission completed.\n")

utils/prompts.py

@@ -116,11 +116,15 @@ def generate_paper_prompts(paper_info, section):
     # ref_instruction_subprompt - give AI references
     # self_subprompt - give AI existing written parts
     # output_subprompt - tell AI how to output
-    fundamental_subprompt = "Your task is to write the {section} section of the machine learning paper with the title '{title}'. {description}\n"
+    fundamental_subprompt = "Your task is to write the {section} section of the paper with the title '{title}'. {description}\n"
     instruction_subprompt = "\n" \
                             "Your response should follow the following instructions:\n" \
                             "{instruction}\n" \
                             "- Start with \section{{{section}}}\n"
+
+    abstract_instruction_subprompt = "\n" \
+                            "Your response should follow the following instructions:\n" \
+                            "{instruction}\n"
     ref_instruction_subprompt = "- Read references. " \
                                 "Every time you use information from the references, you need to appropriately cite it (using \citep or \citet)." \
                                 "For example of \citep, the sentence where you use information from lei2022adaptive \citep{{lei2022adaptive}}. " \
@@ -141,7 +145,7 @@ def generate_paper_prompts(paper_info, section):
         template=fundamental_subprompt + instruction_subprompt + self_subprompt + output_subprompt)
     abstract_prompts = PromptTemplate(
         input_variables=["title", "description", "instruction", "section", "paper"],
-        template=fundamental_subprompt + instruction_subprompt + self_subprompt + abstract_output_subprompt)
+        template=fundamental_subprompt + abstract_instruction_subprompt + self_subprompt + abstract_output_subprompt)
 
     if section in ["introduction", "related works", "backgrounds"]:
         # title + references + instruction

utils/tex_processing.py

@@ -1,4 +1,6 @@
 import os
+import re
+import shutil
 
 def replace_title(save_to_path, title):
     # Define input and output file names
@@ -25,5 +27,46 @@ def replace_title(save_to_path, title):
 
 # return all .png and replace it using placeholder.
 
+def find_tex_files(directory_path):
+    tex_files = []
+
+    for filename in os.listdir(directory_path):
+        if filename.endswith(".tex"):
+            tex_files.append(filename)
+
+    return tex_files
+
+def find_figure_names(tex_file_path):
+    # Regular expression pattern to find \includegraphics commands
+    pattern = r'\\includegraphics.*?{(.*?)}'
+    with open(tex_file_path, 'r') as file:
+        content = file.read()
+    # Find all matches in the file content
+    matches = re.findall(pattern, content)
+    # Matches will be a list of figure names
+    return matches
+
+def create_copies(output_dir):
+    tex_files = find_tex_files(output_dir)
+    for tex_file in tex_files:
+        path =  os.path.join(output_dir, tex_file)
+        all_figs = find_figure_names(path)
+        for fig in all_figs:
+            original_fig = os.path.join(output_dir, "fig.png")
+            target_fig = os.path.join(output_dir, fig)
+            shutil.copy2(original_fig, target_fig)
+
+
+
+
+
+
+if __name__ == "__main__":
+    auto_draft = os.path.dirname(os.getcwd())
+    directory_path = "outputs/outputs_20230520_110413/"
+    path = os.path.join(auto_draft, directory_path)
+
+    create_copies(path)
+