Update app.py

app.py

@@ -1,17 +1,19 @@
-# config.py
 import os
+import streamlit as st
+import openai
+from llama_index.core import SimpleDirectoryReader, VectorStoreIndex, StorageContext
+from llama_index.core.node_parser import SentenceSplitter
+from llama_index.retrievers.bm25 import BM25Retriever
+from llama_index.core.retrievers import BaseRetriever
 
+# Configuration
 class Config:
     OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
     MODEL_NAME = "gpt-3.5-turbo"
     EMBEDDING_MODEL = "text-embedding-3-small"
     CHUNK_SIZE = 256
 
-# document_processor.py
-from llama_index.core import SimpleDirectoryReader, VectorStoreIndex, StorageContext
-from llama_index.core.node_parser import SentenceSplitter
-import streamlit as st
-
+# Document Processing
 class DocumentProcessor:
     def __init__(self):
         self.splitter = SentenceSplitter(chunk_size=Config.CHUNK_SIZE)
@@ -31,10 +33,7 @@ class DocumentProcessor:
         storage_context.docstore.add_documents(nodes)
         return VectorStoreIndex(nodes=nodes, storage_context=storage_context), nodes
 
-# retriever.py
-from llama_index.retrievers.bm25 import BM25Retriever
-from llama_index.core.retrievers import BaseRetriever
-
+# Hybrid Retriever
 class HybridRetriever(BaseRetriever):
     def __init__(self, vector_retriever, bm25_retriever):
         self.vector_retriever = vector_retriever
@@ -51,9 +50,7 @@ class HybridRetriever(BaseRetriever):
                 node_ids.add(n.node.node_id)
         return all_nodes
 
-# llm_service.py
-import openai
-
+# LLM Service
 class LLMService:
     def __init__(self, model_name):
         self.model_name = model_name
@@ -87,13 +84,7 @@ class LLMService:
         )
         return response.choices[0].message.content
 
-# app.py
-import streamlit as st
-from config import Config
-from document_processor import DocumentProcessor
-from retriever import HybridRetriever
-from llm_service import LLMService
-
+# Main Application Class
 class PromptOptimizationApp:
     def __init__(self):
         self.doc_processor = DocumentProcessor()