removed Azure

app.py

@@ -1,33 +1,20 @@
-from dotenv import load_dotenv
-from llama_index import (
-    GPTVectorStoreIndex, SimpleDirectoryReader,
-    ServiceContext, LLMPredictor, PromptHelper,
-    StorageContext, load_index_from_storage
-)
 from langchain.chat_models import ChatOpenAI
+from llama_index import StorageContext, load_index_from_storage
+from dotenv import load_dotenv
+import openai
 import gradio as gr
 import os
 
-load_dotenv()
-OPENAI_API_KEY=os.getenv("OPENAI_API_KEY")
-AZURE_BASE_URL=os.getenv("AZURE_BASE_URL")
-AZURE_DEPLOYMENT_NAME=os.getenv("AZURE_DEPLOYMENT_NAME")
-AZURE_OPENAI_API_KEY=os.getenv("AZURE_OPENAI_API_KEY")
-AZURE_OPENAI_API_VERSION=os.getenv("AZURE_OPENAI_API_VERSION")
-
 def chatbot(input_text):
     storage_context = StorageContext.from_defaults(persist_dir="./storage")
     index = load_index_from_storage(storage_context)
-
     qe = index.as_query_engine()
     response = qe.query(input_text)
-
     return response.response
 
-
+load_dotenv()
 iface = gr.Interface(fn=chatbot,
                      inputs=gr.components.Textbox(lines=7, label="Enter your text"),
                      outputs="text",
                      title="Custom-trained AI Chatbot")
-
 iface.launch()

create_index.py

@@ -1,51 +1,12 @@
 from dotenv import load_dotenv
-from llama_index import (
-    GPTVectorStoreIndex, SimpleDirectoryReader,
-    ServiceContext, LLMPredictor, PromptHelper,
-)
-from langchain.chat_models import ChatOpenAI
+from llama_index import GPTVectorStoreIndex, SimpleDirectoryReader
 import os
 
-load_dotenv()
-OPENAI_API_KEY=os.getenv("OPENAI_API_KEY")
-AZURE_BASE_URL=os.getenv("AZURE_BASE_URL")
-AZURE_DEPLOYMENT_NAME=os.getenv("AZURE_DEPLOYMENT_NAME")
-AZURE_OPENAI_API_KEY=os.getenv("AZURE_OPENAI_API_KEY")
-AZURE_OPENAI_API_VERSION=os.getenv("AZURE_OPENAI_API_VERSION")
-
 def construct_index(directory_path):
-    max_input_size = 4096
-    num_outputs = 512
-    max_chunk_overlap = 20
-    chunk_size_limit = 600
-
-    prompt_helper = PromptHelper(
-        max_input_size, num_outputs, max_chunk_overlap, chunk_size_limit=chunk_size_limit
-    )
-
-    llm_predictor = LLMPredictor(
-        #llm=ChatOpenAI(temperature=0.7, model_name="gpt-3.5-turbo", max_tokens=num_outputs)
-
-        llm = AzureChatOpenAI(
-            openai_api_base=AZURE_BASE_URL,
-            openai_api_version=AZURE_OPENAI_API_VERSION,
-            deployment_name=AZURE_OPENAI_DEPLOYMENT_NAME,
-            openai_api_key=AZURE_OPENAI_API_KEY,
-            openai_api_type = "azure",
-        )
-    )
-
-    service_context = ServiceContext.from_defaults(
-        llm_predictor=llm_predictor, prompt_helper=prompt_helper
-    )
-
     documents = SimpleDirectoryReader(directory_path).load_data()
-
-    index = GPTVectorStoreIndex.from_documents(
-        documents, service_context=service_context
-    )
+    index = GPTVectorStoreIndex.from_documents(documents)
     index.storage_context.persist()
-
     return index
 
+load_dotenv()
 index = construct_index("docs")