Upload folder using huggingface_hub

.gitignore

@@ -129,4 +129,6 @@ dmypy.json
 .pyre/
 .idea
 
-test1.ipynb
+test1.ipynb
+
+app1.py

app.py

@@ -145,9 +145,10 @@ def generate_unique_string():
     unique_string = hashlib.sha256((mac_str + cpu_info).encode()).hexdigest()
     return unique_string
 
-def llm_response(query):
+def llm_response(query, session_id):
     titles, links, res_titles, res_links = [], [], [], []
-    unique_id = generate_unique_string()
+    print(session_id)
+    unique_id = session_id
     config = {"configurable": {"thread_id": unique_id }}
     try:
         # modified_query = runnable.invoke({"input": query, "chat_history": history.messages}).content
@@ -199,9 +200,21 @@ def llm_response(query):
 
 with gr.Blocks() as demo:
     gr.Markdown("## FORTIFIED AI Assistant!")
-    gr.Markdown("### I'll try to answer any questions related to FORTIFIED program. Tell me what's on your mind?")
-    with gr.Row():
+    with gr.Row(visible=True) as role_selector:
+        with gr.Column(scale=4):
+            pass
+        with gr.Column(scale=2):
+            gr.Markdown("### Choose your role:")
+            homeowner = gr.Button("Homeowner", variant="primary")
+            design_pro = gr.Button("Design Professionals", variant="primary")
+            evaluator = gr.Button("Evaluator", variant="primary")
+        with gr.Column(scale=4):
+            pass
+    with gr.Row(visible=False) as chat_container:
         with gr.Column():
+            session_id = gr.Textbox(label="Session ID", visible=False)
+            gr.Markdown(
+                "### I'll try to answer any questions related to FORTIFIED program. Tell me what's on your mind?")
             chatbot = gr.Chatbot(type="messages", height=400)
             msg = gr.Textbox(label="Hit the Enter to send your question", placeholder="What's on your mind?", show_copy_button=True)
             send = gr.Button("Send")
@@ -210,16 +223,31 @@ with gr.Blocks() as demo:
                 return "", history + [{"role": "user", "content": user_message}]
 
 
-            def bot(history: list):
-                bot_message = llm_response(history[-1]['content'])
+            def bot(history: list, session_id_i):
+                bot_message = llm_response(history[-1]['content'], session_id_i)
                 history.append({"role": "assistant", "content": ""})
                 for character in bot_message:
                     history[-1]['content'] += character
                     yield history
 
-
-            msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(bot, chatbot, chatbot)
-            send.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(bot, chatbot, chatbot)
+            msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+                bot, [chatbot, session_id], chatbot
+            )
+            send.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+                bot, [chatbot, session_id], chatbot
+            )
+
+    def start():
+        unique_id = uuid.uuid4()
+
+        return {
+            chat_container: gr.update(visible=True),
+            role_selector: gr.update(visible=False),
+            session_id: gr.update(value=unique_id),
+        }
+    homeowner.click(start, [], [chat_container, role_selector, session_id])
+    design_pro.click(start, [], [chat_container, role_selector,session_id])
+    evaluator.click(start, [], [chat_container, role_selector, session_id])
 
 if __name__ == "__main__":
     demo.launch()