add package resource mapping

.gitignore

@@ -0,0 +1,2 @@
+
+.DS_Store

app.py

@@ -7,7 +7,7 @@ import gradio as gr
 
 from swiftsage.agents import SwiftSage
 from swiftsage.utils.commons import PromptTemplate, api_configs, setup_logging
-
+from pkg_resources import resource_filename
 
 def solve_problem(problem, max_iterations, reward_threshold, swift_model_id, sage_model_id, feedback_model_id, use_retrieval, start_with_sage):
     # Configuration for each LLM
@@ -30,7 +30,8 @@ def solve_problem(problem, max_iterations, reward_threshold, swift_model_id, sag
     }
 
     # specify the path to the prompt templates
-    prompt_template_dir = './swiftsage/prompt_templates'
+    # prompt_template_dir = './swiftsage/prompt_templates'
+    prompt_template_dir = resource_filename('swiftsage', 'prompt_templates')
     dataset = [] 
     embeddings = [] # TODO: for retrieval augmentation (not implemented yet now)
     s2 = SwiftSage(