Update app.py

app.py

@@ -32,7 +32,7 @@ def humanize(
     )
     return result
 
-def ai_generated_test(text):
+def ai_generated_test_polygraf(text):
     url = "http://34.66.10.188/ai-vs-human"
     access_key = "6mcemwsFycVVgVjMFwKXki3zJka1r7N4u$Z0Y|x$gecC$hdNtpQf-SpL0+=k;u%BZ" 
     headers = {
@@ -56,11 +56,11 @@ def ai_generated_test_sapling(text):
 
 def ai_check(text, option):
     if option == 'Polygraf AI':
-        return ai_generated_test(text)
+        return ai_generated_test_polygraf(text)
     elif option == 'Sapling AI':
         return ai_generated_test_sapling(text)
-    else :
-        return ai_generated_test(text)
+    else:
+        return ai_generated_test_polygraf(text)
 
 def update_visibility_api(model):
     if model in ['OpenAI GPT 3.5', 'OpenAI GPT 4']:
@@ -98,7 +98,7 @@ with gr.Blocks() as demo:
                 choices=[
                     "Polygraf AI",
                     "Sapling AI",
-                ],label="Select AI Detector", visible=False)
+                ],label="Select AI Detector", value="Polygraf AI", visible=False)
             only_ai_check_btn = gr.Button("AI Check", visible= False)
             bcLabel = gr.Label(label="Source", visible= False)
             humanizer_btn = gr.Button("Humanize", visible=False)
@@ -164,7 +164,7 @@ with gr.Blocks() as demo:
         ai_article_btn.click(on_first_button_click, inputs=None, outputs=ai_generated)
         ai_article_btn.click(on_first_button_click, inputs=None, outputs=ai_label)
         ai_article_btn.click(on_first_button_click, inputs=None, outputs=only_ai_check_btn)
-        # only_ai_check_btn.click(on_first_button_click, inputs=None, outputs=ai_detector_dropdown)
+        ai_article_btn.click(on_first_button_click, inputs=None, outputs=ai_detector_dropdown)
         only_ai_check_btn.click(on_first_button_click, inputs=None, outputs=bcLabel)
         only_ai_check_btn.click(on_first_button_click, inputs=None, outputs=model_dropdown)
         only_ai_check_btn.click(on_first_button_click, inputs=None, outputs=humanizer_btn)
@@ -192,15 +192,15 @@ with gr.Blocks() as demo:
 
 
         only_ai_check_btn.click(
-            fn=ai_generated_test,
-            inputs=[ai_label],
+            fn=ai_check,
+            inputs=[ai_label, ai_detector_dropdown],
             outputs=[bcLabel],
             api_name="ai_check",
     )
         
         only_ai_check_btn2.click(
-            fn=ai_generated_test,
-            inputs=[output_label],
+            fn=ai_check,
+            inputs=[ai_label, ai_detector_dropdown],
             outputs=[bcLabel2]
     )