Separating functions

app.py

@@ -1,25 +1,29 @@
 import gradio as gr
 
+def main_function(input_text, target_level, action):
+    if action == "Calculate Input Level":
+        return calc_input_reading_level(input_text), None, None, None
+    elif action == "Generate Text":
+        output_text, output_level, similarity = get_output_text_level_and_similarity(input_text, target_level)
+        return None, output_text, output_level, similarity
+
 def calc_input_reading_level(input_text):
-    # This is a placeholder function. Replace it with the actual implementation.
+    # Placeholder function implementation.
     return len(input_text) % 10  # Random operation as a placeholder
 
 def get_output_text_level_and_similarity(input_text, target_level):
-    # This is a placeholder function. Replace it with the actual implementation.
+    # Placeholder function implementation.
     output_text = input_text[::-1]  # Example operation: reverse the input
     output_level = int(target_level)  # Assuming target_level is directly the output level
     similarity = 0.5  # Example fixed similarity
     return output_text, output_level, similarity
 
-# Create the Gradio app interface
 iface = gr.Interface(
-    fn={
-        "Input Reading Level": calc_input_reading_level,
-        "Output Text, Level and Similarity": get_output_text_level_and_similarity
-    },
+    fn=main_function,
     inputs=[
         gr.components.Textbox(label="Input Text", lines=2),
-        gr.components.Dropdown(choices=["1", "2", "3", "4", "5"], label="Target Reading Level")
+        gr.components.Dropdown(choices=["1", "2", "3", "4", "5"], label="Target Reading Level"),
+        gr.components.Radio(choices=["Calculate Input Level", "Generate Text"], label="Action")
     ],
     outputs=[
         gr.components.Text(label="Input Reading Level"),
@@ -27,8 +31,8 @@ iface = gr.Interface(
         gr.components.Text(label="Output Reading Level"),
         gr.components.Text(label="Output Text Similarity")
     ],
-    title="Text Reading Level and Generation",
-    description="This app calculates the reading level of an input text and generates new text at a specified reading level, along with its reading level and similarity."
+    title="Text Analysis Tool",
+    description="Calculate the reading level of input text or generate and analyze new text at a specified reading level."
 )
 
 iface.launch()