Update app.py

app.py

@@ -19,7 +19,6 @@ wahl_o_mat_data = load_json_data('data/wahl-o-mat.json')
 
 def predict(
     api_key,
-    user_input,
     model1,
     model2,
     prompt_manipulation,
@@ -31,7 +30,7 @@ def predict(
     ):
                 
     print("Ideology Test:", ideology_test)
-    print("Political Statement Number:", political_statement)
+    print(political_statement)
     
     if prompt_manipulation == "Impersonation (direct steering)":
         prompt = f"""[INST] Du bist ein Politiker der Partei {direct_steering_option}. {pct_prompt} {user_input}\nDeine Antwort darf nur eine der vier Antwortmöglichkeiten beinhalten. [/INST]"""
@@ -104,7 +103,8 @@ def main():
                 # Initialize 'political_statement' with default 'Wahl-O-Mat' values
                 political_statement_initial_choices = [f"{i+1}. {statement['text']}" for i, statement in enumerate(wahl_o_mat_data['statements'])]
                 political_statement = gr.Dropdown(
-                    label="Selecet political statement or enter you own",
+                    label="Select political statement or enter you own",
+                    value="1. Auf allen Autobahnen soll ein generelles Tempolimit gelten.", # default value
                     choices=political_statement_initial_choices,  # Set default to 'Wahl-O-Mat' statements
                     allow_custom_value = True
                 )
@@ -135,16 +135,8 @@ def main():
                 # Link the dropdowns so that the option dropdown updates based on the selected prompt manipulation
                 prompt_manipulation.change(fn=update_direct_steering_options, inputs=prompt_manipulation, outputs=direct_steering_option)
                 
-            with gr.Row():
-                gr.Markdown("#### OR:")
-    
-    
-            with gr.Row():
-                user_input1 = gr.Textbox(label="Prompt", placeholder="Enter your own political statement here")
                 
             with gr.Row():
-                api_key_input = gr.Textbox(label="API Key", placeholder="Enter your API key here", show_label=True, type="password")
-                user_input = gr.Textbox(label="Prompt", placeholder="Enter your message here")
                 model_selector1 = gr.Dropdown(label="Model 1", choices=["mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mixtral-8x22B-Instruct-v0.1"])
                 model_selector2 = gr.Dropdown(label="Model 2", choices=["mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mixtral-8x22B-Instruct-v0.1"])
                 submit_btn = gr.Button("Submit")
@@ -156,22 +148,17 @@ def main():
 
         with gr.Tab("Settings"):
             with gr.Column():
-                api_key_input = gr.Textbox(label="API Key", type="password")
+                api_key_input = gr.Textbox(label="API Key", placeholder="Enter your API key here", show_label=True, type="password")
                 temp_input = gr.Slider(minimum=0, maximum=1, step=0.01, label="Temperature", value=0.7)
                 top_p_input = gr.Slider(minimum=0, maximum=1, step=0.01, label="Top P", value=1)
         
         # Link settings to the predict function
         submit_btn.click(
             fn=predict,
-            inputs=[api_key_input, user_input, model_selector1, model_selector2, prompt_manipulation, direct_steering_option, ideology_test, political_statement, temp_input, top_p_input],
+            inputs=[api_key_input, model_selector1, model_selector2, prompt_manipulation, direct_steering_option, ideology_test, political_statement, temp_input, top_p_input],
             outputs=[output1, output2]
         )
-        # submit_btn.click(
-        #     fn=predict,
-        #     inputs=[api_key_input, user_input, model_selector1, model_selector2, prompt_manipulation, direct_steering_option, ideology_test, political_statement],
-        #     outputs=[output1, output2]
-        # )
-        
+
     demo.launch()
 
 if __name__ == "__main__":