Update app.py

app.py

@@ -121,6 +121,20 @@ def toxicityAnalys():
         return {"status": "pass", "toxicity": predicted_sentiment}
     except Exception as e: return {"status": "error", "details": { "error_code": 123, "error_details": str(e).replace("\n", " | ") }} , 400
 
+############## 
+# GENERATE TEXT
+generator = pipeline("text-generation", model="IlyaGusev/rugpt_medium_turbo_instructed")
+@app.route('/text/api/v1/generate', methods=['POST'])
+def generateText():
+    try:
+        text = helpers.getFromRequest(request, "text")
+        if not text: return {"status": "error", "details": { "error_code": 101, "error_details": "No text provided" }} , 400
+        result = generator(text, max_length=1024, return_full_text=False, do_sample=bool(randint(0,1)), temperature=float(randint(5,15))/10.0)
+        return {"status": "pass", "result": result[0]["generated_text"]} 
+    except Exception as e: return {"status": "error", "details": { "error_code": 123, "error_details": str(e).replace("\n", " | ") }} , 400
+
+##############
+# INIT APP
 if __name__ == "__main__":
     config = configFile()
     with open(config['config-path'], "w", encoding="utf-8") as outfile: