JirasakJo commited on
Commit
17fb98b
·
verified ·
1 Parent(s): 31784af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -5
app.py CHANGED
@@ -89,15 +89,31 @@ def load_custom_css():
89
  """, unsafe_allow_html=True)
90
 
91
  def initialize_pipeline():
92
- """Initialize RAG pipeline with configurations"""
93
  try:
94
  openai_api_key = os.getenv('OPENAI_API_KEY') or st.secrets['OPENAI_API_KEY']
95
- config = create_default_config(openai_api_key)
96
- config.localization.enable_thai_normalization = True
97
- config.retriever.top_k = 5
98
- config.model.temperature = 0.3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
99
  pipeline = AcademicCalendarRAG(config)
100
 
 
101
  with open("calendar.json", "r", encoding="utf-8") as f:
102
  calendar_data = json.load(f)
103
  pipeline.load_data(calendar_data)
 
89
  """, unsafe_allow_html=True)
90
 
91
  def initialize_pipeline():
92
+ """Initialize RAG pipeline with enhanced configurations"""
93
  try:
94
  openai_api_key = os.getenv('OPENAI_API_KEY') or st.secrets['OPENAI_API_KEY']
95
+
96
+ # Create detailed configuration
97
+ config = PipelineConfig(
98
+ model=ModelConfig(
99
+ openai_api_key=openai_api_key,
100
+ embedder_model="sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
101
+ openai_model="gpt-4",
102
+ temperature=0.3
103
+ ),
104
+ retriever=RetrieverConfig(top_k=5),
105
+ cache=CacheConfig(
106
+ enabled=True,
107
+ cache_dir=Path("./cache"),
108
+ ttl=86400
109
+ ),
110
+ processing=ProcessingConfig(batch_size=32),
111
+ localization=LocalizationConfig(enable_thai_normalization=True)
112
+ )
113
+
114
  pipeline = AcademicCalendarRAG(config)
115
 
116
+ # Load calendar data
117
  with open("calendar.json", "r", encoding="utf-8") as f:
118
  calendar_data = json.load(f)
119
  pipeline.load_data(calendar_data)