Update app.py

app.py

@@ -10,37 +10,7 @@ client = AsyncOpenAI(
 )
 
 assistantID = os.getenv("OPENAI_ASSISTANT_ID")
-username = os.getenv("YOUR_ID")
-password = os.getenv("YOUR_PASSWORD")
-
-mytitle = "<h1 align=center>RTL AI News Reader : What happened in the country 🇱🇺 and in the world 🌎 ?</h1>"
-
-mydescription="""
-<h3 align='center'>Which topic interests you : 🐶 🏃🏻‍♂️ 🌗 🍇 🌈 🍽️ 🏆 🚘 ✈️ 🩺 </h3>
-<table width=100%>
-  <tr>
-    <th width=50% bgcolor="Moccasin">Ask your questions in english or another language :</th>
-    <th bgcolor="Khaki">Response from the OpenAI File-Search Assistant :</th>
-  </tr>
-</table>
-"""
-
-myarticle ="""
-<h3>Background :</h3>
-<p>This HuggingFace Space demo was created by <a href="https://github.com/mbarnig">Marco Barnig</a>. As an artificial intelligence,
-the <a href="https://platform.openai.com/docs/models">OpenAI model</a> gpt-4o-mini-2024-07-18 is used via API,
-which can utilize up to 128,000 tokens as context, provide an answer to a question with a maximum of 16,384 tokens, 
-and process up to 200,000 tokens per minute (TPM). All english content from RTL.lu from the beginning up to September 2024 has been split into 16 JSON files
-and uploaded to a Vector Store by the OpenAI File-Search Assistant "RTL English News Reader." 
-Each file contains fewer than 5 million tokens, which is an upper limit for the AI model. It is possible to upload up to 10,000 files to an OpenAI Assistant.
-The responses of the examples are cached and therefore displayed without delay.</p>
-"""
-
-myinput = gr.Textbox(lines=3, label=" What would you like to know ?")
-
-myexamples = [
-"What happened in 2014 ?"
-]
+mypassword = os.getenv("RTL_PASSWORD")
 
 class EventHandler(AsyncAssistantEventHandler):
     def __init__(self) -> None:
@@ -86,14 +56,17 @@ async def initialize_thread():
     session_data["thread_id"] = thread.id
 
 async def generate_response(user_input):
-    assistant_id = session_data["assistant_id"]
-    thread_id = session_data["thread_id"]
-
-    # Add a Message to the Thread
-    oai_message = await client.beta.threads.messages.create(
-        thread_id=thread_id,
-        role="user",
-        content=user_input
+    if user_input == "":
+        yield "Schreif eng Fro als Input"
+    else:
+        assistant_id = session_data["assistant_id"]
+        thread_id = session_data["thread_id"]
+
+        # Add a Message to the Thread
+        oai_message = await client.beta.threads.messages.create(
+            thread_id=thread_id,
+            role="user",
+            content=user_input
     )
 
     # Create and Stream a Run
@@ -111,34 +84,59 @@ async def generate_response(user_input):
             yield event_handler.response_text
 
 # Gradio interface function (generator)
-async def gradio_chat_interface(user_input):
-    # Create a new event loop if none exists (or if we are in a new thread)
-    try:
-        loop = asyncio.get_running_loop()
-    except RuntimeError:
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-
-    # Initialize the thread if not already done
-    if session_data["thread_id"] is None:
-        await initialize_thread()
-
-    # Generate and yield responses
-    async for response in generate_response(user_input):
-        yield response
-
-# Set up Gradio interface with streaming
-interface = gr.Interface(
-    fn=gradio_chat_interface,
-    inputs=myinput,
-    outputs="markdown",
-    title=mytitle,
-    description=mydescription,
-    article=myarticle,
-    live=False,
-    allow_flagging="never",
-    examples=myexamples
-)
-
-# Launch the Gradio app
-interface.launch(auth=(username, password), auth_message="<h1>RTL AI News Reader</h1><p>This HuggingFace Space is a prototype and is not yet accessible to everyone. The project is based on a file search assistant using OpenAI's API and employs the GPT-4o-mini model. You need to use a Chrome browser. AI specialists interested in accessing it can request a username and password by contacting [email protected].</p>")
+async def gradio_chat_interface(mode, password, user_input, example):
+    if mode == "Beispiller":
+        filename = example[-6:-2] + ".md"
+        file = open("examples/" + filename, "r")
+        output = file.read()
+        yield output
+    else:
+        # check the password
+        if password == "":
+            yield "To serach you need to enter an RTL password !"
+        elif password != mypassword:
+            yield "Please enter the correct RTL password !"
+        else:
+
+            # Create a new event loop if none exists (or if we are in a new thread)
+            try:
+                loop = asyncio.get_running_loop()
+            except RuntimeError:
+                loop = asyncio.new_event_loop()
+                asyncio.set_event_loop(loop)
+
+            # Initialize the thread if not already done
+            if session_data["thread_id"] is None:
+                await initialize_thread()
+
+            # Generate and yield responses
+            async for response in generate_response(user_input):
+                yield response
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        myTitle = gr.HTML("<h2 align=center>RTL AI News Reader : What happened in the country 🇱🇺 or in the world 🌎 ?</h2>")   
+    with gr.Row():
+        myDescription = gr.HTML("""
+            <h3 align='center'>Wat fir een Thema interesséiert Iech ?</h3>
+            <p align='center'>🐶 🏃🏻‍♂️ 🌗 🍇 🌈 🍽️ 🏆 🚘 ✈️ 🩺 </p>
+            <p align='center' bgcolor="Moccasin">Submit your question in english or in another language !</p>
+            """            
+        )
+    with gr.Row():
+        mode = gr.Radio(choices=["Search", "Examples"], label = "You can run the examples without password.", value = "Examples")
+        pw = gr.Textbox(lines=1, label="Enter the correct RTL password !")
+    with gr.Row():
+            question = gr.Textbox(lines=3, label="Please submit your question ?")
+    with gr.Row():    
+            examples = gr.Radio(["Wat war lass am Juni 2023 ?", "Wat ass gewosst iwwert de SREL ?", "Wat fir eng Katastroph war 2022 zu Lëtzebuerg ?", "Koumen an de leschte Jore gréisser Kriminalfäll viru Geriicht ?"], value="Wat ass gewosst iwwert de SREL ?" , label="Beispiller")
+    with gr.Row():
+        clear = gr.Button("Clear")
+        submit = gr.Button("Submit")
+    with gr.Row():
+        mySubtitle = gr.HTML("<p align='center' bgcolor='Khaki'>English RTL News :</p>")
+    with gr.Row():
+        myOutput = gr.Markdown(label="Answer from the OpenAI File-Search Assistent :")
+
+    submit.click(fn = gradio_chat_interface, inputs=[mode, pw, question, examples], outputs = myOutput)
+demo.launch()