Upload folder using huggingface_hub

.gitignore

@@ -1,2 +1,2 @@
-private
+private
 .DS_Store

app.py

@@ -1,54 +1,44 @@
-from transformers import pipeline, Conversation
+from openai import OpenAI
 import gradio as gr
-from datetime import datetime
-
-from transformers.utils import logging
-logging.set_verbosity_error()
-
-# Info about ['google/gemma-2b-it'](https://huggingface.co/google/gemma-2b-it)
-chatbot = pipeline(task="conversational",
-                   model="./models/google/gemma-2b-it")
-
-def get_job_description(company_name, sector, job_position):
-    start_time = datetime.now()
-
-    messages = [
-            {
-                "role": "user",
-                "content": f"""
-                    You are the Talent Acquisition Manager at { company_name }, a company of the { sector } sector.
-                    Your goal is to hire people in different departments of the company.
-                    I want to generate a Job Description for the role '{ job_position }' in a creative, modern, formal and attractive way.
-                    """
-            },
-            {
-                "role": "assistant",
-                "content": "Description:"
-            }
-        ]
-    
-    conversation = Conversation(messages)
-    conversation = chatbot(conversation, do_sample=True, max_new_tokens=600, temperature=1)
-    response = conversation.messages[-1]["content"]
-    
-    delta_time = datetime.now() - start_time
-
-    return response, delta_time
+
+openai_apikey = "sk-proj-JW9oMGO1QmwnOxQ3vUA8T3BlbkFJO2zo3KLERLeds7HJmt6y"
+
+client = OpenAI(api_key = openai_apikey)
+
+thread = client.beta.threads.create()
+
+def get_answer(question):
+    message = client.beta.threads.messages.create(
+        thread_id = thread.id,
+        role = "user",
+        content = question
+    )
+
+    run = client.beta.threads.runs.create_and_poll(
+        thread_id = thread.id,
+        assistant_id = "asst_LgX8lE9erZR3ktgzOvkWNAQ8",
+        instructions = "Your name is Bossard ChatGPT. You enable access to Bossard technical knowledge in an interactive and targeted way. You act like an experienced Bossard engineer and can advise with skillful hints in the entirety and provide relevant further information. I want the answer you give to be more or less 100 words maximum if necessary. Try responding with a list, a short answer or a step-by-step guide if you feel like you should give an answer like that for the question."
+    )
+        
+    messages = client.beta.threads.messages.list(
+            thread_id = thread.id
+    )
+
+    response = messages.data[0].content[0].text.value
+
+    return response
 
 with gr.Blocks() as demo:
     gr.Markdown(
     """
-    # Job Description generator
-    ### This SLM demo uses ['google/gemma-2b-it'](https://huggingface.co/google/gemma-2b-it)
+    # Bossard ChatGPT
+    ### PoC with OpenAI GPT4o
     """)
 
     interface = gr.Interface(
-        fn = get_job_description,
-        inputs = [gr.Textbox(placeholder="Enter the company name", label="Company name"),
-                  gr.Textbox(placeholder="Enter the company sector", label="Company sector"),
-                  gr.Textbox(placeholder="Enter the job position", label="Job position")],
-        outputs = [gr.Markdown(),
-               gr.Textbox(label="Elapsed time")]
+        fn = get_answer,
+        inputs = [gr.Textbox(placeholder="Enter the question", label="Question")],
+        outputs = [gr.Markdown()]
     )
 
 demo.launch(server_port=7860)

requirements.txt

@@ -1,3 +1,2 @@
-transformers
-torch
+openai
 gradio