Update calendar_rag.py

calendar_rag.py

@@ -968,56 +968,56 @@ class AcademicCalendarRAG:
                 "query_info": {}
             }
     
-def main():
-    """Main function for processing real calendar queries"""
-    try:
-        # Load API key
-        with open("key.txt", "r") as f:
-            openai_api_key = f.read().strip()
-        
-        # Use create_default_config instead of direct PipelineConfig initialization
-        config = create_default_config(openai_api_key)
-    
-        # Customize config for Thai academic calendar use case
-        config.localization.enable_thai_normalization = True
-        config.retriever.top_k = 5  # Adjust based on your needs
-        config.model.temperature = 0.3  # Lower temperature for more focused responses
-        
-        # Initialize pipeline with enhanced config
-        pipeline = AcademicCalendarRAG(config)
-        
-        # Load calendar data
-        with open("calendar.json", "r", encoding="utf-8") as f:
-            calendar_data = json.load(f)
-        pipeline.load_data(calendar_data)
-        
-        # Real queries to process
-        queries = input in web
-        
-        print("Processing calendar queries...")
-        print("=" * 80)
-        
-        for query in queries:
-            result = pipeline.process_query(query)
-            print(f"\nQuery: {query}")
-            print(f"Answer: {result['answer']}")
-            
-            # # Print retrieved documents for verification
-            # print("\nRetrieved Documents:")
-            # for i, doc in enumerate(result['documents'], 1):
-            #     print(f"\nDocument {i}:")
-            #     print(doc.content)
-            
-            # # Print query understanding info
-            # print("\nQuery Understanding:")
-            # for key, value in result['query_info'].items():
-            #     print(f"{key}: {value}")
+# def main():
+#     """Main function for processing real calendar queries"""
+#     try:
+#         # Load API key
+#         with open("key.txt", "r") as f:
+#             openai_api_key = f.read().strip()
+        
+#         # Use create_default_config instead of direct PipelineConfig initialization
+#         config = create_default_config(openai_api_key)
+    
+#         # Customize config for Thai academic calendar use case
+#         config.localization.enable_thai_normalization = True
+#         config.retriever.top_k = 5  # Adjust based on your needs
+#         config.model.temperature = 0.3  # Lower temperature for more focused responses
+        
+#         # Initialize pipeline with enhanced config
+#         pipeline = AcademicCalendarRAG(config)
+        
+#         # Load calendar data
+#         with open("calendar.json", "r", encoding="utf-8") as f:
+#             calendar_data = json.load(f)
+#         pipeline.load_data(calendar_data)
+        
+#         # Real queries to process
+#         queries = input in web
+        
+#         print("Processing calendar queries...")
+#         print("=" * 80)
+        
+#         for query in queries:
+#             result = pipeline.process_query(query)
+#             print(f"\nQuery: {query}")
+#             print(f"Answer: {result['answer']}")
+            
+#             # # Print retrieved documents for verification
+#             # print("\nRetrieved Documents:")
+#             # for i, doc in enumerate(result['documents'], 1):
+#             #     print(f"\nDocument {i}:")
+#             #     print(doc.content)
+            
+#             # # Print query understanding info
+#             # print("\nQuery Understanding:")
+#             # for key, value in result['query_info'].items():
+#             #     print(f"{key}: {value}")
                 
-            print("=" * 80)
+#             print("=" * 80)
             
-    except Exception as e:
-        logger.error(f"Pipeline execution failed: {str(e)}")
-        raise
+#     except Exception as e:
+#         logger.error(f"Pipeline execution failed: {str(e)}")
+#         raise
 
-if __name__ == "__main__":
-    main()
+# if __name__ == "__main__":
+#     main()