Update file paths and variable names

src/one-shot/generate_annotations.py

@@ -97,7 +97,7 @@ def annotate_dialogues(df: pd.DataFrame, api_url: str) -> pd.DataFrame:
         pd.DataFrame: The DataFrame with added annotations.
     """
     for index, row in tqdm(df.iterrows(), total=df.shape[0], desc="Annotating Dialogues"):
-        prompt_original = llama_prompt_no_schema(row['Complete Dialogue'])
+        prompt_original = llama_prompt_no_schema(row['Original Dialogue'])
         response_original = post_http_request(prompt_original, api_url, n=1, stream=False)
         prompt_generated = llama_prompt_no_schema(row['utterances_joined'])
         response_generated = post_http_request(prompt_generated, api_url, n=1, stream=False)
@@ -142,10 +142,10 @@ def main(args):
         args: Command line arguments.
     """
     api_url = f"http://{args.host}:{args.port}/generate"
-    input_file = f"../../data/own_data/dialogues/{args.input_name}"
+    input_file = f"../../data/own_data/one-shot/dialogues/{args.input_name}"
     df = pd.read_csv(input_file, sep=',', quoting=csv.QUOTE_NONE, escapechar='/')
     df = annotate_dialogues(df, api_url)
-    output_file = f"../../data/own_data/dialogues/{args.output_name}"
+    output_file = f"../../data/own_data/one-shot/dialogues/{args.output_name}"
     df.to_csv(output_file, sep=',', index=False, quoting=csv.QUOTE_NONE, escapechar='/')
 
 if __name__ == "__main__":