update app.py

app.py

@@ -7,7 +7,7 @@ from openai import OpenAI
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
 # Function to process the input and generate the output
-def process_query(wiki_page, embed_dim, query, api_key=None, mode="OpenAI"):
+def process_query(wiki_page, embed_dim, query, mode, api_key=None):
     model_mapping = {
         "Arabic-mpnet-base-all-nli-triplet": "Omartificial-Intelligence-Space/Arabic-mpnet-base-all-nli-triplet",
         "Arabic-all-nli-triplet-Matryoshka": "Omartificial-Intelligence-Space/Arabic-all-nli-triplet-Matryoshka",
@@ -63,35 +63,32 @@ def process_query(wiki_page, embed_dim, query, api_key=None, mode="OpenAI"):
 
     return "\n\n".join([f"Model: {model_name}\nResponse: {response}" for model_name, response in responses.items()])
 
-# Define the interface
-wiki_page_input = gr.Textbox(label="Wikipedia Page (in Arabic)")
-query_input = gr.Textbox(label="Query (in Arabic)")
-api_key_input = gr.Textbox(label="OpenAI API Key", type="password", visible=False)
-
-embed_dim_choice = gr.Dropdown(
-    choices=[768, 512, 256, 128, 64],
-    label="Embedding Dimension"
-)
-
-mode_choice = gr.Radio(
-    choices=["OpenAI", "OpenSource"],
-    label="Choose Mode"
-)
-
-output_text = gr.Textbox(label="Output")
-
-def on_mode_change(mode):
-    if mode == "OpenAI":
-        api_key_input.visible = True
-    else:
-        api_key_input.visible = False
-
-mode_choice.change(on_mode_change, inputs=mode_choice, outputs=api_key_input)
-
-gr.Interface(
-    fn=process_query,
-    inputs=[wiki_page_input, embed_dim_choice, query_input, api_key_input, mode_choice],
-    outputs=output_text,
-    title="Arabic Wiki RAG",
-    description="Choose a Wikipedia page, embedding model, and dimension to answer a query in Arabic."
-).launch()
+with gr.Blocks() as demo:
+    wiki_page_input = gr.Textbox(label="Wikipedia Page (in Arabic)")
+    query_input = gr.Textbox(label="Query (in Arabic)")
+    api_key_input = gr.Textbox(label="OpenAI API Key", type="password", visible=False)
+
+    embed_dim_choice = gr.Dropdown(
+        choices=[768, 512, 256, 128, 64],
+        label="Embedding Dimension"
+    )
+
+    mode_choice = gr.Radio(
+        choices=["OpenAI", "OpenSource"],
+        label="Choose Mode"
+    )
+
+    output_text = gr.Textbox(label="Output")
+
+    def on_mode_change(mode):
+        if mode == "OpenAI":
+            api_key_input.visible = True
+        else:
+            api_key_input.visible = False
+
+    mode_choice.change(on_mode_change, inputs=mode_choice, outputs=api_key_input)
+
+    submit_button = gr.Button("Submit")
+    submit_button.click(process_query, inputs=[wiki_page_input, embed_dim_choice, query_input, mode_choice, api_key_input], outputs=output_text)
+
+demo.launch()