Update app.py

app.py

@@ -22,8 +22,37 @@ def sentiment_analysis(text: str) -> dict:
     }
 
 
-# Create the Gradio interface
-demo = gr.Interface(
+def prime_factorization(n: int) -> dict:
+    """
+    Compute the prime factorization of a number.
+
+    Args:
+        n (int): Positive integer
+
+    Returns:
+        dict: Prime factors and their exponents (with string keys for JSON compatibility)
+    """
+    if n <= 0:
+        return {"error": "Please enter a positive integer."}
+
+    i = 2
+    factors = {}
+    while i * i <= n:
+        while n % i == 0:
+            factors[i] = factors.get(i, 0) + 1
+            n //= i
+        i += 1
+    if n > 1:
+        factors[n] = 1
+
+    # Convert integer keys to strings
+    str_factors = {str(k): v for k, v in factors.items()}
+
+    return {"factors": str_factors}
+
+
+# Define Gradio interfaces for each function
+sentiment_interface = gr.Interface(
     fn=sentiment_analysis,
     inputs=gr.Textbox(placeholder="Enter text to analyze..."),
     outputs=gr.JSON(),
@@ -31,6 +60,20 @@ demo = gr.Interface(
     description="Analyze the sentiment of text using TextBlob"
 )
 
+factorization_interface = gr.Interface(
+    fn=prime_factorization,
+    inputs=gr.Number(value=1, label="Number", precision=0),
+    outputs=gr.JSON(),
+    title="Prime Factorization",
+    description="Compute the prime factorization of a number"
+)
+
+# Combine into a tabbed interface
+demo = gr.TabbedInterface(
+    interface_list=[sentiment_interface, factorization_interface],
+    tab_names=["Sentiment Analysis", "Prime Factorization"]
+)
+
 # Launch the interface and MCP server
 if __name__ == "__main__":
     demo.launch(mcp_server=True)