maybe works

app.py

@@ -5,15 +5,35 @@ import requests
 API_KEY = "your_huggingface_api_key"
 
 def unified_function(operation, input_text):
-    if operation == "Text Generation":
-        return "gen" + input_text
-
-    elif operation == "Sentiment Analysis":
-
-        return "sent" + input_text
-
-    elif operation == "Echo Text":
-        return "echo" + input_text
+    if operation == "Clean Diff":
+        added_lines = []
+        for line in input_text.split('\n'):
+            if line.startswith('+') and not line.startswith('+++'):
+                added_line = line[1:]  # Remove the '+' sign
+                if sum(len(l) + 1 for l in added_lines) + len(added_line) <= 2000:
+                    added_lines.append(added_line)
+                else:
+                    return "too long, try again"
+                    break
+        return '\n'.join(added_lines)
+
+    elif operation == "Classify":
+        API_URL = "https://api-inference.huggingface.co/models/davidgaofc/TechDebtClassifier"
+        headers = {"Authorization": f"Bearer {API_KEY}"}
+        data = {"inputs": input_text}
+
+        response = requests.post(API_URL, headers=headers, json=data)
+        result = response.json()
+        return result
+
+    elif operation == "Generate Label":
+        API_URL = "https://api-inference.huggingface.co/models/davidgaofc/TechDebtLabeler"
+        headers = {"Authorization": f"Bearer {API_KEY}"}
+        data = {"inputs": input_text}
+
+        response = requests.post(API_URL, headers=headers, json=data)
+        result = response.json()
+        return result
 
 def huggingface_login(api_key):
     API_KEY = api_key
@@ -22,7 +42,7 @@ def huggingface_login(api_key):
 interface = gr.Interface(
     fn=unified_function,
     inputs=[
-        gr.Dropdown(["Text Generation", "Sentiment Analysis", "Echo Text"], label="Select Operation"),
+        gr.Dropdown(["Clean Diff", "Classify", "Generate Label"], label="Select Operation"),
         gr.Textbox(label="Input Text")
     ],
     outputs="text",