Upload 5 files

.env

@@ -0,0 +1 @@
+OPENAI_API_KEY='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'

.gitignore

@@ -0,0 +1,3 @@
+venv/
+.env
+__pycache__/

gradio_ui.py

@@ -0,0 +1,13 @@
+import gradio as gr
+
+from utils import generate_response
+
+with gr.Blocks() as demo:
+
+    chatbot = gr.Chatbot(label='Network Design Chatbot', height=600)
+    msg = gr.Textbox()
+    clear = gr.ClearButton([msg, chatbot])
+    
+    msg.submit(generate_response, [msg, chatbot], [msg, chatbot])
+
+demo.launch()

requirements.txt

@@ -0,0 +1,72 @@
+aiofiles==23.1.0
+aiohttp==3.8.4
+aiosignal==1.3.1
+altair==5.0.1
+annotated-types==0.5.0
+anyio==3.7.1
+async-timeout==4.0.2
+attrs==23.1.0
+certifi==2023.5.7
+charset-normalizer==3.2.0
+click==8.1.4
+contourpy==1.1.0
+cycler==0.11.0
+exceptiongroup==1.1.2
+fastapi==0.100.0
+ffmpy==0.3.0
+filelock==3.12.2
+fonttools==4.40.0
+frozenlist==1.3.3
+fsspec==2023.6.0
+gradio==3.36.1
+gradio_client==0.2.8
+h11==0.14.0
+httpcore==0.17.3
+httpx==0.24.1
+huggingface-hub==0.16.4
+idna==3.4
+importlib-resources==6.0.0
+Jinja2==3.1.2
+jsonschema==4.18.0
+jsonschema-specifications==2023.6.1
+kiwisolver==1.4.4
+linkify-it-py==2.0.2
+markdown-it-py==2.2.0
+MarkupSafe==2.1.3
+matplotlib==3.7.2
+mdit-py-plugins==0.3.3
+mdurl==0.1.2
+multidict==6.0.4
+numpy==1.25.1
+openai==0.27.8
+orjson==3.9.2
+packaging==23.1
+pandas==2.0.3
+Pillow==10.0.0
+pydantic==2.0.2
+pydantic_core==2.1.2
+pydub==0.25.1
+Pygments==2.15.1
+pyparsing==3.0.9
+python-dateutil==2.8.2
+python-dotenv==1.0.0
+python-multipart==0.0.6
+pytz==2023.3
+PyYAML==6.0
+referencing==0.29.1
+requests==2.31.0
+rpds-py==0.8.10
+semantic-version==2.10.0
+six==1.16.0
+sniffio==1.3.0
+starlette==0.27.0
+toolz==0.12.0
+tqdm==4.65.0
+typing_extensions==4.7.1
+tzdata==2023.3
+uc-micro-py==1.0.2
+urllib3==2.0.3
+uvicorn==0.22.0
+websockets==11.0.3
+yarl==1.9.2
+zipp==3.16.0

utils.py

@@ -0,0 +1,51 @@
+import os
+import random
+import time
+
+import openai
+from dotenv import load_dotenv, find_dotenv
+
+load_dotenv(find_dotenv())
+
+openai.api_key = os.getenv('OPENAI_API_KEY')
+
+def chat_completion(messages: list) -> str:
+    try:
+        completion = openai.ChatCompletion.create(
+            model='gpt-3.5-turbo',
+            messages=messages
+        )
+        return completion['choices'][0]['message']['content']
+    except:
+        return 'We are facing a technical issue at this moment.'
+
+def generate_messages(messages: list, query: str) -> list:
+    formated_messages = [
+        {
+            'role': 'system',
+            'content': 'You are a helpful Network Design assistant.'
+        }
+    ]
+    for m in messages:
+        formated_messages.append({
+            'role': 'user',
+            'content': m[0]
+        })
+        formated_messages.append({
+            'role': 'assistant',
+            'content': m[1]
+        })
+    formated_messages.append(
+        {
+            'role': 'user',
+            'content': query
+        }
+    )
+    return formated_messages
+
+def generate_response(query: str, chat_history: list) -> tuple:
+        messages = generate_messages(chat_history, query)
+        bot_message = chat_completion(messages)
+        chat_history.append((query, bot_message))
+        time.sleep(random.randint(0, 5))
+        return '', chat_history