Upload 2 files

app.py

@@ -1,75 +1,41 @@
-# Imports
 import gradio as gr
 from helper_functions import *
 
 with gr.Blocks() as app:
-    gr.Markdown('# FundedNext Customer Service Chatbot')
-    session_data = gr.State([
-        [{"role": "system", "content": pre_text}],[]
-    ])
-    def user(user_message, history):
-        return "", history + [[user_message, None]]
+    gr.Markdown('# Prompt Generator for FundedNext')
+    session_data = gr.State([{"current_system_prompt": ""}])
 
-    def bot(history, session_data_fn):
-        messages_archived = session_data_fn[0]
-        messages_current = session_data_fn[1]
-        bot_message, messages_archived, messages_current = get_reply(history[-1][0], messages_archived, messages_current)
-        history[-1][1] = bot_message
-        session_data_fn[0] = messages_archived
-        session_data_fn[1] = messages_current
-        return history, session_data_fn
+    def get_prompt(user_message, session_data):
+        if(user_message == ""):
+            return_message = "Please enter your message"
+            return return_message
+        pre_text = session_data[0]["current_system_prompt"]
+        return_message = pre_text + "\n\n" + get_context(user_message)
+        return return_message
+    def set_pre_text(system_prompt, session_data):
+        session_data[0]["current_system_prompt"] = system_prompt
+        return session_data
     
-    def reset_memory(session_data_fn):
-        messages_archived = session_data_fn[0]
-        # print("Message Archived Len=", len(messages_archived))
-        if(len(messages_archived)>=21):
-            messages_archived = messages_archived[0:1] + messages_archived[3:]     
-            session_data_fn[0] = messages_archived
-        return session_data_fn
+    with gr.Tab("Generate Prompt"):
+        user_message = gr.Textbox(label = "Enter your message")
+        prompt = gr.Textbox(label="Generated Prompt", interactive=True, lines=20)
+    with gr.Tab("Edit System Prompt"):
+        system_prompt = gr.Textbox(
+            label="System Prompt", interactive=True, lines=15
+        )
+        gr.Markdown("## System Prompt Examples")
+        gr.Examples(
+            examples = [[pre_text]],
+            inputs = [system_prompt]
+        )
 
-    def clear_data(session_data_fn):
-        messages_archived = [
-        {"role": "system", "content": pre_text}
-        ]   
-        messages_current = []   
-        session_data_fn[0] = messages_archived
-        session_data_fn[1] = messages_current
-        return None, session_data_fn
-
-    def get_context_gr(session_data_fn):
-        messages_current = session_data_fn[1]
-        return str(messages_current)
-
-    with gr.Tab("Chat"):
-        with gr.Row():
-            with gr.Column():
-                msg = gr.Textbox()
-                with gr.Row():
-                    submit = gr.Button("Submit")
-                    clear = gr.Button("Clear")
-            with gr.Column():
-                chatbot = gr.Chatbot()
-            
-    with gr.Tab("Prompt"):
-        context = gr.Textbox()
-        submit_p = gr.Button("Check Prompt")
-    # Tab Chat
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, [chatbot, session_data], [chatbot, session_data]
-    ).then(
-        fn = reset_memory, inputs = session_data, outputs = session_data
-    )
-    submit.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, [chatbot, session_data], [chatbot, session_data]
-    ).then(
-        fn = reset_memory, inputs = session_data, outputs = session_data
+    user_message.submit(
+        fn = get_prompt, inputs = [user_message, session_data], outputs = prompt
+    ).then(lambda:"", inputs=None, outputs=user_message)
+    system_prompt.change(
+        fn = set_pre_text, inputs = [system_prompt, session_data], outputs = session_data
     )
-    clear.click(
-        fn = clear_data, 
-        inputs = session_data, 
-        outputs = [chatbot, session_data],
-        queue = False
-    )
-    # Tab Prompt
-    submit_p.click(get_context_gr, session_data, context, queue=False)
-app.launch(debug=True)
+    
+
+app.launch(auth=(os.getenv("id"), os.getenv("password")), show_api=False)
+# app.launch()

helper_functions.py

@@ -44,24 +44,3 @@ def get_context(query):
 My question is: “{query}”
 """
 
-def get_reply(message, messages_archived, messages_current):
-    
-    if message:
-        messages_current = messages_archived.copy()
-        context = get_context(message)
-        messages_current.append(
-            {"role": "user", "content": context}
-        )
-        chat = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo", messages=messages_current, temperature=0
-        )
-    
-        reply = chat.choices[0].message.content
-        messages_archived.append({"role": "user", "content": message})
-        messages_archived.append({"role": "assistant", "content": reply})
-    # If no message is provided, return a string that says "No Message Received"
-    else:
-        reply = "No Message Received"
-
-    return reply, messages_archived, messages_current
-