Upload app.py

app.py

@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Created on Tue Apr 25 04:44:31 2023
+
+@author: mai2125
+"""
+
+from gpt_index import (
+    SimpleWebPageReader, 
+    WikipediaReader, 
+    GPTListIndex, 
+    GPTSimpleVectorIndex, 
+    LLMPredictor, 
+    QuestionAnswerPrompt,
+    PromptHelper
+)
+from langchain.chat_models import ChatOpenAI
+from langchain.prompts.chat import ChatPromptTemplate, SystemMessagePromptTemplate
+import gradio as gr
+import docx2txt
+import sys
+import os
+# openai api key
+os.environ["OPENAI_API_KEY"] = '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'
+
+def chatbot(input_text):
+    # preset prompt
+    
+    query_str = "What state is Atlanta in?"
+    QA_PROMPT_TMPL = (
+        "You are an AI specialized in Atlanta.\n"
+        "We have provided context information below. \n"
+        "---------------------\n"
+        "{context_str}"
+        "\n---------------------\n"
+        "Given this information, please answer the question: {query_str}\n"
+        "If this question definitely does not involve Atlanta, say I am a chatbot designed to answer queries about specifically Atlanta.\n"
+    )
+    QA_PROMPT = QuestionAnswerPrompt(QA_PROMPT_TMPL)
+    # Takes in the input from the user to deliver responses
+    index = GPTSimpleVectorIndex.load_from_disk('index.json')
+    response = index.query(input_text, text_qa_template = QA_PROMPT)
+    return response.response
+
+# creates the interface for the user to interact with the chatbot
+iface = gr.Interface(fn=chatbot,
+                     inputs=gr.components.Textbox(lines=7, label="Enter your text"),
+                     outputs="text",
+                     title="PD4 Chatbot based on Atlanta Wiki")
+
+iface.launch()