Update app.py

app.py

@@ -47,32 +47,27 @@ def initialize_vector_store_index(data_path, service_context):
     loader = PyPDFLoader("./Cloudflare.pdf")
     pages = loader.load_and_split()
     # Load the index from a file
-    
+
     #index = VectorStoreIndex.from_documents(documents, service_context=service_context)
     embeddings_2 = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
     documents = SimpleDirectoryReader("./").load_data()
     index = VectorStoreIndex.from_documents(documents, service_context=service_context)
     return index
 
-def main():
-    st.title("Cloudflare RAG")
-        # User input
-    user_input = st.text_input("Enter your message:")
 
-    # Configure and initialize components
-    llm = configure_llama_model()
-    embed_model = configure_embeddings()
-    service_context = configure_service_context(llm, embed_model)
-    index = initialize_vector_store_index("./", service_context)
+
+# Configure and initialize components
+llm = configure_llama_model()
+embed_model = configure_embeddings()
+service_context = configure_service_context(llm, embed_model)
+index = initialize_vector_store_index("./", service_context)
+
+    # User input
+user_input = input()
 
 
+query_engine = index.as_query_engine()
+response = query_engine.query(user_input)
+print(response)
 
-    if user_input:
-        # Generate response
-        query_engine = index.as_query_engine()
-        response = query_engine.query(user_input)
-        # Display response
-        st.text_area("ChatGPT Response:", response, height=400)
 
-if __name__ == "__main__":
-    main()